search

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏
Apache License 2.0
1.5k stars 446 forks source link

关于负样本代码labels = np.zeros((0,5), dtype=np.float32) #110

Open JieShare opened 3 years ago

JieShare commented 3 years ago

是不是写错了，应该是labels = np.zeros((1,5), dtype=np.float32)？代表负样本的标签为0，ground truth的坐标和高宽都为0？

zbyuan commented 3 years ago

0表示0个正样本，1就是有一个了 当然其实都没什么关系，都影响不了大局

JieShare commented 3 years ago

0表示0个正样本，1就是有一个了 当然其实都没什么关系，都影响不了大局

不好意思我还是有疑问。如果我将没有目标的背景图片作为负样本(也就是说对应的txt为空)加入训练集，那么这张图片在代码中对应的label是labels = np.zeros((0,5), dtype=np.float32)吗？那labels不是为[]了吗？这还怎么进行损失函数的计算呢？