1.构造coco数据集,file_name为具体的ndarray文件名,类名的改变和class_num的配置和之前一样。保存的npy文件是归一化之后的结果
2.修改数据加载代码,将
2.修改网络输入inchannel为npy的通道数,此处采用的backbone是resnext,npy通道数为4,应该修改的地方是
3.修改均值和方差分别为[0, 0, 0, 0]和[1, 1, 1, 1],list长度与npy文件的通道数一样,由于之前已做过归一化,此处相当于没有变化(减0除1)
4.删除pretrain行
再次运行,即可正常训练(建议epoch数量参考scratch的配置)
过去已逝,未来太远,只争今朝