73,719
社区成员
发帖
与我相关
我的任务
分享数据集是乳腺癌肿瘤图像,使用的是FCN-ResNet50预训练模型。
本来想自定义一个网络,但是之前从来没有使用过预训练模型,今天试一试,看看效果怎么样。
一般对医学处理图像的研究,预训练模型是采用Mask-GCNN,虽然数据集中有mask图像,但是开始遇到了一点小Bug:该模型需要有锚框。我当时有些发懵,遇到问题后第一时间换模型(*^_^*),后来思考了一下,可能跟Mask-GCNN是同时进行图像分割和图像分类的模型有关,所以如果大家以后确实想用Mask-GCNN,可以在自定义数据集类的target字典中,将boxes设置为全0张量即可,这个思路我已经尝试过了,没有什么问题,只不过设备受限,我还没有进行完整的训练。
使用U-Net进行乳腺癌图像分割_Rvosuke的博客-CSDN博客
这个标题写为U-Net其实是由于FCN-ResNet50是U-Net的一种特殊模型(开始我是想自定义Unet的)