Open pprp opened 5 years ago
你好,感谢您的分享,我可以很顺利的训练原版的darknet,但是我在使用你的库的时候遇到了几个问题:
您的库是在训练过程中来对权重文件进行剪枝,还是说直接对已经训练好的权重文件进行剪枝
我使用的是yolov3-spp.cfg文件,但是运行:
./darknet detector train cfg/voc.data cfg/yolov3-spp.cfg yolov3-spp_110000.weights
遇到以下报错:
88 conv 24 1 x 1 / 1 21 x 21 x1024 -> 21 x 21 x 24 89 Type not recognized: [yolo] Unused field: 'mask = 6,7,8' Unused field: 'anchors = 10,13,16,30,33,23,30,61,62,45,59,119,116,90,156,198,373,326' Unused field: 'classes = 3' Unused field: 'num = 9' Unused field: 'jitter = .3' Unused field: 'ignore_thresh = .7' Unused field: 'truth_thresh = 1' Unused field: 'random = 1' 90 route 86 91 conv 256 1 x 1 / 1 21 x 21 x 512 -> 21 x 21 x 256 92 Type not recognized: [upsample] Unused field: 'stride = 2' 93 route 92 61 94 Layer before convolutional layer must output image.: Cannot allocate memory darknet: ./src/utils.c:224: error: Assertion `0' failed. Aborted (core dumped)
因此我猜测,您没有实现upsample yolo等层,对吗?
之后我又尝试了yolov3-tiny.cfg,也遇到相似的问题
所以想问问您具体修改的情况,非常感谢(ps: 如果能出教程就更好了:happy)
如果能分享一下 tiny-yolo.cfg 中的内容就更好了
你好,感谢您的分享,我可以很顺利的训练原版的darknet,但是我在使用你的库的时候遇到了几个问题:
您的库是在训练过程中来对权重文件进行剪枝,还是说直接对已经训练好的权重文件进行剪枝
我使用的是yolov3-spp.cfg文件,但是运行:
遇到以下报错:
因此我猜测,您没有实现upsample yolo等层,对吗?
之后我又尝试了yolov3-tiny.cfg,也遇到相似的问题
所以想问问您具体修改的情况,非常感谢(ps: 如果能出教程就更好了:happy)