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的配置)

posted on 2023-01-05 17:18  一点飞鸿  阅读(51)  评论(0编辑  收藏  举报