用CenterNet對旋轉(zhuǎn)目標(biāo)進(jìn)行檢測

極市導(dǎo)讀
?作者對CenterNet進(jìn)行了部分修改,以此進(jìn)行旋轉(zhuǎn)目標(biāo)檢測。本文介紹了這種工作的代碼、數(shù)據(jù)處理和樣例,供讀者參考。>>加入極市CV技術(shù)交流群,走在計(jì)算機(jī)視覺的最前沿
前言

代碼
train.py:模型的訓(xùn)練
predict.py:模型的前向推理
backbone:模型的主干網(wǎng),給了DLA和ResNet的DCN與普通版本,層數(shù)可以自定義
loss.py:模型的損失函數(shù)
dataset.py:模型的數(shù)據(jù)處理
數(shù)據(jù)處理
如何對旋轉(zhuǎn)目標(biāo)打標(biāo)簽? 如何將角度信息添加到VOC格式標(biāo)簽? 如何計(jì)算旋轉(zhuǎn)IOU指標(biāo)? 其它
Demo






推薦閱讀

評論
圖片
表情
