來源：投稿 作者：LSC
編輯：學(xué)姐

最終成績是:68.46473 第二名

比賽網(wǎng)址

http://challenge.xfyun.cn/topic/info?type=helmet-wear&option=ssgy

賽題任務(wù)

在本次比賽中需要參賽選手搭建計(jì)算機(jī)視覺模型識別出照片中的安全帽位置。

本次賽題包括三類目標(biāo)物體：Helmet，Person，Head，訓(xùn)練集4千張圖片，測試集1千張圖片。訓(xùn)練集數(shù)據(jù)集標(biāo)注格式為：

評估指標(biāo)

本次競賽的評價(jià)標(biāo)準(zhǔn)采用mAP(mean Average Precision)準(zhǔn)確率指標(biāo)，最高分為1。計(jì)算方法參考代碼參考：https://github.com/Cartucho/mAP

識別結(jié)果文件詳細(xì)說明：

(1)標(biāo)簽順序需要與測試集文本保持一致；

(2)具體格式如下：

提交對應(yīng)文件名以txt結(jié)尾的文件結(jié)果文件，如hard_hat_workers4007.txt，提交文件內(nèi)容的格式為：

baseline代碼我是在autodl平臺上運(yùn)行的，我是用PaddleDetection框架來進(jìn)行檢測的，要先將數(shù)據(jù)集和PaddleDetection.zip上傳到平臺

(1)首先是解壓數(shù)據(jù)集和安裝所需環(huán)境

(2)將標(biāo)簽文件由txt格式轉(zhuǎn)成xml格式，再轉(zhuǎn)為json格式，見train1.ipynb中的代碼，主要是運(yùn)行makexml函數(shù)和PaddleDetection/tools/x2coco.py模塊

(3)修改PaddleDetection中的'PaddleDetection/configs/ppyoloe/ppyoloe_plus_crn_x_80e_coco.yml'文件及其相關(guān)文件'../datasets/coco_detection.yml', '../runtime.yml', './_base_/optimizer_80e.yml','./_base_/ppyoloe_plus_crn.yml', './_base_/ppyoloe_plus_reader.yml'等，重點(diǎn)是coco_detection.yml的文件路徑，ppyoloe_plus_crn_x_80e_coco.yml里面的pretrain_weights，是預(yù)訓(xùn)練模型權(quán)重這是修改后ppyoloe_plus_crn_x_80e_coco.yml的內(nèi)容。

這是修改后的coco_detection.yml的內(nèi)容

修改后的runtime.yml的內(nèi)容:

修改后的optimizer_80e.yml的內(nèi)容:

(4)運(yùn)行如下命令行代碼，開始訓(xùn)練模型

全部數(shù)據(jù)訓(xùn)練和驗(yàn)證

是用所有訓(xùn)練集數(shù)據(jù)進(jìn)行訓(xùn)練和驗(yàn)證，epoch是20，但是到了第10輪效果沒有高效地提升，停止了訓(xùn)練，保存最好的模型。

(5)運(yùn)行如下命令行代碼，推理測試集數(shù)據(jù)

其中weights參數(shù)是最好的訓(xùn)練模型的權(quán)重的絕對路徑，output_dir是結(jié)果輸出絕對路徑，infer_dir是測試集圖片數(shù)據(jù)的絕對路徑，最終生成推理結(jié)果的圖片和測試集的推理結(jié)果bbox.json都存儲在autodl-tmp/infer_output2/路徑下。

(6)對autodl-tmp/infer_output2/bbox.json數(shù)據(jù)進(jìn)行讀取和處理，將每張測試集圖片的結(jié)果寫在同名的txt文本下，存儲在/root/detection-results目錄中。

這個(gè)時(shí)候要注意生成的bbox不知道對應(yīng)的原圖是哪個(gè)，所以對PaddelDetection的代碼要稍微做修改，在PaddleDetection/ppdet/engine/trainer.py代碼的第841行后加入:

(7)最后用命令行把/root/detection-results文件夾壓縮下載到本地%cd /root/!tar -cvzf detection-results.tar.gz detection-results解壓再壓縮改為zip文件進(jìn)行提交


比賽資源都安排在“KGMking”公眾號上了！歡迎大家關(guān)注！

baseline＆數(shù)據(jù)集合集都有都有