yolov4 pytorch修改的地方記錄

2021-10-25 14:57:40 字數 451 閱讀 8324

1.get_map.py

目的:修改pr曲線的文字顯示大小

位置:945,951, 952

plt.xlabel(

"recall"

, fontsize=16)

plt.ylabel(

"precision"

, fontsize=16)

#32改為16

注意力機制的pytorch-yolov4

2. eval_voc.py

(1)–mode det 引數:生成detetion-result資料夾,該資料夾要手動建立

(2) --mode val 引數:生成out資料夾, 裡面都是***.txt檔案,用於後續的map計算

get_map.py

執行該檔案後,在根木中自動生成result資料夾, pr曲線等都是儲存在result資料夾下。

YOLOv3 Pytorch學習筆記

五月一直埋頭鑽研faster r cnn,但苦於電腦不支援gpu,一直連個簡單的結果都沒跑出來 期間還掙扎著安裝cuda,結果就是ubuntu系統一崩再崩 yolo官網 翻譯 按官網上的要求一步一步走,就可得到如下檢測結果 如果想訓練自己的資料,可參考部落格 yolov3 訓練自己的資料,講的非常詳...

yolov3 yolov4與yolov5效能對比

yolov5s網路最小,速度最少,ap精度也最低。但如果檢測的以大目標為主,追求速度,倒也是個不錯的選擇。其他的三種網路,在此基礎上,不斷加深加寬網路,ap精度也不斷提公升,但速度的消耗也在不斷增加。目前使用下來,yolov5s的模型十幾m大小,速度很快,線上生產效果可觀,嵌入式裝置可以使用。在相同...

yolo v3 pytorch 模型訓練筆記

annotations 標註資訊 分為train xml和val xml datasets 資料集和label 分為train images val images train labels val labels ptj.data ptj.names 類別名 ptj test 做測試用的集 resul...