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

mAP计算差异问题 #58

Open ZhangWurui opened 4 years ago

ZhangWurui commented 4 years ago

您好，我成功使用您的工程对我自己的数据集（VOC格式）训练的YOLOv3模型，进行了稀疏化、剪枝和微调的操作，并且训练以及剪枝过程中都显示mAP高达0.96左右（和我正常训练得到的模型差不多）。 L MMC}E3HTOKE0R2Y)PY3XE

之后我用得到模型，在 https://github.com/AlexeyAB/darknet 中使用 ./darknet detector map data/my.data cfg/my.cfg weights/my.weights 命令计算mAP，发现最多只能达到0.86左右。 UFU9GXE(6DIYNY7O(LFA5V3

然后我又使用您的工程采用不同的策略进行了剪枝等操作，显示得到的mAP都比我在 https://github.com/AlexeyAB/darknet 里计算的高不少。

请问是这两个工程对mAP的计算方式有差异，还是我有什么地方疏忽了。万分感谢！ （刚开始接触这些东西，请多指教了）

zbyuan commented 4 years ago

AB版本的yolo在数据的优化方式、损失函数、设置的周期参数等等有U版本的yolo都略有差异 这是正常的

ZhangWurui commented 4 years ago

AB版本的yolo在数据的优化方式、损失函数、设置的周期参数等等有U版本的yolo都略有差异 这是正常的

好的，多谢