目錄

代碼獲取

map原理

map提高技巧

技巧總結：

實戰：

1、效果不佳map55.55%

1）單獨調整get_dr_txt.py中的self.iou = 0.3

2）單獨調整get_map,py中的minoverlap:

3)同時調整minoverlap和self.iou

---

本文是在faster_rcnn模型的基礎上進行map的提高

代碼獲取

map計算代碼clone于：

這個是用來繪制mAP曲線的。
https://github.com/Cartucho/mAP
這個是用來獲取繪制mAP曲線所需的txt的
https://github.com/bubbliiiing/count-mAP-txt

map原理

map計算原理參考：https://blog.csdn.net/weixin_44791964/article/details/104695264?utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-2.control&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-2.control

map提高技巧

map提高技巧參考：https://blog.csdn.net/qq_28664681/article/details/108232353

技巧總結：

1、訓練階段：
1）增大學習率
2）減小batchsize2、驗證階段：
1）調小get_dr_txt.py中的置信度，這就相當與將ap曲線往后延長，自然就增大了
2）增大get_dr_txt.py中的nms階段的iou_threshold
3)減小get_map.py中的minoverlap

實戰：

1、效果不佳map55.55%

調整參數使得map增加

訓練相關參數

lr = 1e-4

Batch_size = 2

Init_Epoch = 0

Freeze_Epoch = 50

Confidence = 0.8

Iou_thre = 0.3

1）單獨調整get_dr_txt.py中的self.iou = 0.3

Map = 60.5%

提高了5%

?

2）單獨調整get_map,py中的minoverlap:

Minoverlap = 0.1

Iou = 0.5

Map = 66.49%

提高11%

3)同時調整minoverlap和self.iou

Minoverlap = 0.1

Self.iou = 0.2

Map = 69.02%

提高13.5%