1. 程式人生 > >mask-RCNN筆記——coco安裝及使用

mask-RCNN筆記——coco安裝及使用

mask-rcnn的coco資料集使用:

coco安裝:

1、下載程式碼,clone或者直接download

https://github.com/waleedka/coco

2、我使用的是python,進入目錄,並執行make

進入有makefile的資料夾

在終端開啟,執行命令make,

將make後的pycocotools複製到Mask_RCNN下

3.下載預訓練的coco權重檔案(mask_rcnn_coco.h5)【4】

https://github.com/matterport/Mask_RCNN/releases

coco資料集:

1、下載coco資料【8】

wget http://images.cocodataset.org/annotations/image_info_test2014.zip

wget http://images.cocodataset.org/annotations/annotations_trainval2014.zip

wget http://images.cocodataset.org/zips/test2014.zip

wget http://images.cocodataset.org/zips/val2014.zip

wget http://images.cocodataset.org/zips/train2014.zip

得到壓縮包,並解壓

2、資料集的使用:

在coco資料夾下建資料夾dataset

其下annotations資料夾中放圖片資料夾的json檔案

其下放置圖片資料夾(train2014是圖片資料夾)

參考程式碼:

【8】https://blog.csdn.net/u014734886/article/details/78830713

【4】https://blog.csdn.net/wei_guo_xd/article/details/78579534