win10 faster rcnn 用自己的数据进行训练【二】

第一步:制作自己的数据

参考:https://blog.csdn.net/sinat_30071459/article/details/50723212

第二步:修改的文件

在  pascal_voc.py 文件中将将self._classes 改成自己的类别

在demo.py文件中将

   net.create_architecture(sess, "TEST", 2,
                            tag='default', anchor_scales=[8, 16, 32])

其中的2改成你的类别总数

记住:这里标签不能用大写 要不然会出现  keyError


可能出现的问题:

anchor_target_layer.py”, line 137, in forward 

gt_argmax_overlaps = overlaps.argmax(axis=0) ValueError: attempt to get argmax of an empty sequence

解决办法:https://blog.csdn.net/jiajunlee/article/details/50470897

有可能是目标框的宽度和高度,比率,未做标记,图像的宽度和高度

 目标框宽度和高度的差距比较大的去除

matlab代码:https://download.csdn.net/my


参考链接:https://zhuanlan.zhihu.com/p/31772360

                https://blog.csdn.net/some_possible/article/details/79139994

猜你喜欢

转载自blog.csdn.net/j497205974/article/details/80074494