|
本帖最后由 Cat 于 2020-5-4 12:16 编辑
我们不是用的yolov4,而是换了一个简单的双支路输出的网,因为yolov3和yolov4需要预测的输出都是全卷积阵列,而我们的网络最后需要预测一个nx1x1的tensor
所以我们用了avg层或者connected层试图把卷积层输出的CxWxH集合成一个nx1x1的tensor
例子里面的yolov3是可以转换的
以下是yolov3里面出现的所有的层,转换程序都是认的
[convolutional] stride=1~2都行
[upsample]
[route]
[shortcut]
[yolo]
我们自己设计cfg在电脑运行时没有任何问题的,不同之处在于我们在两个输出层之前,多了
[avg]层或者[connected]层
但是似乎转换程序不认识,avg层或者connected层,我们看不到darknet转换程序的.so源码,能否支持一下?
注:因为输入图像不是正方形,如果用卷积层和maxpooling层最后也不一定能得到nx1x1的tensor,所以只能求助于avg层或者connected层 |
|