talebolano / yolov3-network-slimming

yolov3 network slimming剪枝的一种实现
345 stars 93 forks source link

关于在yolov2上使用的问题? #32

Open SyGoing opened 5 years ago

SyGoing commented 5 years ago

我用你的代码在yolov2稀疏化训练时(yolov2 608输入,两类目标), 1)报错提示: 损失非常大,然后崩了,梯度爆炸。还有就是utils.py里的build_targets 提示index=19 in dimension 3 超过边界。 2)源码yolomodel.py中 self.losses["recall"] /= 3 self.losses["precision"] /= 3 为什么要除于3,是因为yolov3有三个输出层么? 那么yolov2的话是不是需要修改呢