yolov5-v3.0嘗鮮

README

20200817, yolov5發(fā)布了3.0擒悬,精度來到了49.2,加上多尺度tta后 50.8


安裝

感覺這個issue, v3.0的release指向的分支有問題,所以保險起見

git clone git@github.com:ultralytics/yolov5.git
git reset --hard 916d4aa

下載weights

bash weights/download_weights.sh

不要用github里面的release來下載猫十,太慢蚓再,用這個比較快

文件布局

yolov5的文件布局我不習(xí)慣,比如data里面放yaml, coco路徑和yolov5同級一姿,有點別扭七咧,我改了一下,在yolov5下新建coco文件夾叮叹,如下艾栋,

coco
|-- data.yaml
`-- model.yaml

model.yaml就是yolov5x.yamldata.yaml改了一下路徑


然后進入data/scripts蛉顽,運行

bash get_coco.sh

會得到data/coco這個文件夾蝗砾,和data.yaml對應(yīng)上了

test

python test.py --data coco/data.yaml --weights weights/yolov5x.pt  --img 640 --conf 0.001 --save-json

可以復(fù)現(xiàn)49.2的map,挺好

水下檢測實戰(zhàn)

python train.py --cfg underwater/model.yaml --data underwater/data.yaml --epochs 100 --weights weights/yolov5x.pt --batch-size 8

5260張 train 2*2080ti, 顯存2*9300mb,6min/epoch悼粮,設(shè)置的100epoch差不多是10h闲勺,有點久
map 0.5, map50 0.847,但是同數(shù)據(jù)的universe 0.523 0.858, cas 64 4d 0.536 0.867扣猫,可能是尺度小了點菜循,

在超算上試一下

python train.py --cfg underwater/model.yaml --data underwater/data.yaml --epochs 100 --weights weights/yolov5x.pt --batch-size 8 --cache-images

2V100(16G) 加上 cache也是6min/epoch,此時是bs8 * 2申尤,換成bs 16 看看速度會不會加快癌幕,換成了bs 16 速度是4min/epoch,還可以昧穿,
bs4 1024 2gpu, 10min/epoch
換成4
V100 bs 6勺远,6不是4的倍數(shù),報了錯粤咪,改成了4*bs4 谚中,還是10min/epoch,總共跑了18h寥枝,map 0.503宪塔,搞了一點點,map50最高是0.845囊拜,欸某筐,垃圾

trouble shooting

進度條給的map低了點

是的, 根據(jù)issue yolov5 test.py里面的評估有點問題冠跷,加上--save-json調(diào)用pycocotools來評估就可以了

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末南誊,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子蜜托,更是在濱河造成了極大的恐慌抄囚,老刑警劉巖,帶你破解...
    沈念sama閱讀 221,576評論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件橄务,死亡現(xiàn)場離奇詭異幔托,居然都是意外死亡,警方通過查閱死者的電腦和手機蜂挪,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,515評論 3 399
  • 文/潘曉璐 我一進店門重挑,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人棠涮,你說我怎么就攤上這事谬哀。” “怎么了严肪?”我有些...
    開封第一講書人閱讀 168,017評論 0 360
  • 文/不壞的土叔 我叫張陵史煎,是天一觀的道長谦屑。 經(jīng)常有香客問我,道長劲室,這世上最難降的妖魔是什么伦仍? 我笑而不...
    開封第一講書人閱讀 59,626評論 1 296
  • 正文 為了忘掉前任结窘,我火速辦了婚禮很洋,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘隧枫。我一直安慰自己喉磁,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 68,625評論 6 397
  • 文/花漫 我一把揭開白布官脓。 她就那樣靜靜地躺著协怒,像睡著了一般。 火紅的嫁衣襯著肌膚如雪卑笨。 梳的紋絲不亂的頭發(fā)上孕暇,一...
    開封第一講書人閱讀 52,255評論 1 308
  • 那天,我揣著相機與錄音赤兴,去河邊找鬼妖滔。 笑死,一個胖子當(dāng)著我的面吹牛桶良,可吹牛的內(nèi)容都是我干的座舍。 我是一名探鬼主播,決...
    沈念sama閱讀 40,825評論 3 421
  • 文/蒼蘭香墨 我猛地睜開眼陨帆,長吁一口氣:“原來是場噩夢啊……” “哼曲秉!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起疲牵,我...
    開封第一講書人閱讀 39,729評論 0 276
  • 序言:老撾萬榮一對情侶失蹤承二,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后纲爸,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體亥鸠,經(jīng)...
    沈念sama閱讀 46,271評論 1 320
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,363評論 3 340
  • 正文 我和宋清朗相戀三年缩焦,在試婚紗的時候發(fā)現(xiàn)自己被綠了读虏。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 40,498評論 1 352
  • 序言:一個原本活蹦亂跳的男人離奇死亡袁滥,死狀恐怖盖桥,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情题翻,我是刑警寧澤揩徊,帶...
    沈念sama閱讀 36,183評論 5 350
  • 正文 年R本政府宣布腰鬼,位于F島的核電站,受9級特大地震影響塑荒,放射性物質(zhì)發(fā)生泄漏熄赡。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,867評論 3 333
  • 文/蒙蒙 一齿税、第九天 我趴在偏房一處隱蔽的房頂上張望彼硫。 院中可真熱鬧,春花似錦凌箕、人聲如沸拧篮。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,338評論 0 24
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽串绩。三九已至,卻和暖如春芜壁,著一層夾襖步出監(jiān)牢的瞬間礁凡,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,458評論 1 272
  • 我被黑心中介騙來泰國打工慧妄, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留顷牌,地道東北人。 一個月前我還...
    沈念sama閱讀 48,906評論 3 376
  • 正文 我出身青樓腰涧,卻偏偏與公主長得像韧掩,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子窖铡,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 45,507評論 2 359