SSD 可以给pascal voc扩展分类吗?
收藏
目前看pascalvoc 提供了20个分类,可以自己增加一个分类吗?比如duck
为什么我一增加分类,训练就出错呢?每个分类下面,图片有数量限制吗?
0
收藏
请登录后评论
请问是Fluid训练吗? SSD可以扩展类别的。PascalVOC 20类,加上背景类,index为0,总共21类。每个类别实际对图片数量没有限制。
请问您是如何增加的?以及错误信息是啥?
在JPEGImages增加一张图片,在Annotations增加一个xml,name字段的值是cs,在Image/Main增加一个cs_trainval.txt文件。
下面交叉熵打印错误
Epoc 0, batch 180, loss 4.66661167145, time 1.24980401993
/paddle/paddle/fluid/operators/math/cross_entropy.cu:29 Assertion `label[i] >= 0 && label[i] < D` failed.
/paddle/paddle/fluid/operators/math/cross_entropy.cu:29 Assertion `label[i] >= 0 && label[i] < D` failed.
/paddle/paddle/fluid/operators/math/cross_entropy.cu:29 Assertion `label[i] >= 0 && label[i] < D` failed.
/paddle/paddle/fluid/operators/math/cross_entropy.cu:29 Assertion `label[i] >= 0 && label[i] < D` failed.
/paddle/paddle/fluid/operators/math/cross_entropy.cu:29 Assertion `label[i] >= 0 && label[i] < D` failed.
/paddle/paddle/fluid/operators/math/cross_entropy.cu:29 Assertion `label[i] >= 0 && label[i] < D` failed.
/paddle/paddle/fluid/operators/math/cross_entropy.cu:29 Assertion `label[i] >= 0 && label[i] < D` failed.
F0917 11:31:21.934497 3499 exception_holder.h:34] Unknown exception caught
*** Check failure stack trace: ***
请问包括背景类,一共不能超过21个分类吗?能在21个分类基础上再增加呢?谢谢!
可以的,可以参考这个文章:https://blog.csdn.net/qq_33200967/article/details/79126830