我在用unet做分割实验时遇到一个问题,由于我们导师给我的数据集只有六张人脸图片,要进行皮肤瑕疵检测,在所有的数据都处理好之后,我将得到的这六张人脸图片对应的json文件和label文件分别分为了两个部分,一部分作为训练集(3张),一部分作为测试集(3张),然后模型训练过程进行顺利,但是模型测试得到的结果却显示nidus值为0,mAP=50%,如下图所示,完全不正常。
我想知道是什么原因导致的,是不是因为我的数据集图片数量太少了呀?还是我划分的训练集和测试集有问题呢?还是什么其他的原因?
训练和测试过程中还有一处报错:UserWarning: floordiv is deprecated, and its behavior will change in a future version of pytorch. It currently rounds toward 0 (like the 'trunc' function NOT 'floor'). ...........
我查了一下是新版pytorch 已经弃用floordiv,要是使用floordiv可以降低pytorch 版本,我感觉好像是没有什么影响的。