Mask Rcnn训练自己的航拍数据集
? ? ? 最近想利用Mask Rcnn訓(xùn)練自己的航拍數(shù)據(jù)(數(shù)據(jù)來(lái)源:自己采集。飛行高度:200m,圖片大小WxH=6000x4000),在進(jìn)行自己的航拍圖片訓(xùn)練時(shí)也與其他同學(xué)進(jìn)行交流,他們一般進(jìn)行識(shí)別的是一類物體,而且圖片不是非常大,基于他們都是在pycharm上能把自己的數(shù)據(jù)集訓(xùn)練成功。到目前為止我還一直在堅(jiān)持用jupyter notebook,至今出現(xiàn)了許多問(wèn)題,目前仍在調(diào)試中:(本人已卡在某一步有半月之久,希望廣大博友能一起討論qq:1753939345).
? ? ?? 現(xiàn)將自己的過(guò)程做一總結(jié):
(1)首先本人將自己采集的航拍圖片拿出來(lái)7張(為了測(cè)試代碼能否跑的通,所以采用少?gòu)垐D片)進(jìn)行做標(biāo)注。標(biāo)注的工具為labelme。采用Anaconda中的Anaconda prompt進(jìn)行l(wèi)abelme的安裝。安裝成功后直接進(jìn)行打開(kāi),便可進(jìn)行圖片的標(biāo)注如下圖。共標(biāo)注3類,分別為:car,lamp,truck。標(biāo)注完的圖片保存之后為1.json格式。如下圖:
(2)在利用mask rcnn進(jìn)行自己的數(shù)據(jù)集訓(xùn)練時(shí),數(shù)據(jù)集的格式要采用coco格式,所以利用label_json_to_data將自己的.json文件進(jìn)行批量轉(zhuǎn)換并保存。本人是在Anaconda prompt中進(jìn)行的轉(zhuǎn)換。在轉(zhuǎn)換時(shí),需進(jìn)入label_json_to_data.exe所在路徑,找到自己的json_to_data.py文件打開(kāi),將
保存路徑進(jìn)行修改,修改為自己轉(zhuǎn)換后的路徑即可。轉(zhuǎn)換方法如下圖所示:
轉(zhuǎn)換完畢后,會(huì)在自己的保存路徑下找到保存的文件,打開(kāi)后會(huì)發(fā)現(xiàn)每一個(gè)保存的文件下都會(huì)有5個(gè)內(nèi)容,即轉(zhuǎn)換完畢。本人的如下所示:Mask RCNN在進(jìn)行訓(xùn)練時(shí)要求的mask數(shù)據(jù)為8位的,所以仍需要檢查自己的label.png是否是8位存儲(chǔ),如若不是需進(jìn)行轉(zhuǎn)換。
(3)經(jīng)過(guò)以上數(shù)據(jù)處理便可進(jìn)行訓(xùn)練。
(4)但是我在訓(xùn)練時(shí)未將自己的數(shù)據(jù)6000x4000大小的進(jìn)行處理(不知是否自己要進(jìn)行裁剪處理,是需要固定尺寸的輸入嗎?這也是自己的疑問(wèn)),所以仍不知自己的原因出在哪里。先將遇到的問(wèn)題進(jìn)行公布,希望能有人指點(diǎn)一下(謝謝)。
?
?
總結(jié)
以上是生活随笔為你收集整理的Mask Rcnn训练自己的航拍数据集的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: LabVIEW设置应用程序显示标签透明
- 下一篇: 解决Mask RCNN自己航拍数据集训练