正如我们上文提到的那样,我们可以通过小样本对SAM模型进行微调,那么我们需要获取有关息肉的数据集,我们应该选用什么数据集呢。
一、kvasir-SEG数据集
市面上和息肉有关的数据集并不太多,比较出名的有如kavasir-SEG数据集
Kvasir-SEG是一个针对结肠息肉像素级分割的内窥镜数据集,举办过MediaEval 2020的比赛。它包括1000张胃肠息肉图像及其对应的分割掩码,这些图像都由资深胃肠病专家亲自标注和核实。官方数据仓库按880:120的比例提供了训练和验证的数据划分,以公正地比较各种方法。此数据集旨在推动结肠息肉的分割、检测、定位和分类方法的研究和创新。
我们可以将kvasir-SEG数据集当作我们的一个数据集。
我们可以进入kvasir-SEG的官网对数据集进行下载。
二、CVC-ClinicDB数据集