mmdetection學習筆記(一) 安裝

2021-10-04 23:22:49 字數 2036 閱讀 2267

配置要求

• linux (windows is not officially supported)

• python 3.5+ (python 2 is not supported)

• pytorch 1.1 or higher

• cuda 9.0 or higher

• nccl 2

• gcc(g++) 4.9 or higher

• mmcv

• os: ubuntu 16.04/18.04 and centos 7.2

• cuda: 9.0/9.2/10.0

• nccl: 2.1.15/2.2.13/2.3.7/2.4.2

gcc(g++): 4.9/5.3/5.4/7.3

安裝步驟

1、建立conda虛擬環境並啟用它

conda create -n open-mmlab python=3.7 -y

source activate open-mmlab

2、安裝pytorch及torchvision

conda install -c pytorch pytorch torchvision -y

3、拷貝mmdetection倉庫

git clone

cd mmdetection

4、安裝requirement.txt及opencv-python

pip install -r requirements.txt

pip install opencv-python

pip install cython

pip install pycocotools

4、編譯mmdetection的配置檔案

python setup.py develop

from mmdet.apis import init_detector

from mmdet.apis import inference_detector

from mmdet.apis import show_result

import cv2

# 模型配置檔案

config_file =

'./configs/faster_rcnn_r50_fpn_1x.py'

# 預訓練模型檔案

checkpoint_file =

'./checkpoints/faster_rcnn_r50_fpn_1x_20181010-3d1b3351.pth'

# 通過模型配置檔案與預訓練檔案構建模型

model = init_detector(config_file, checkpoint_file, device=

'cuda:0'

)# 測試單張並進行展示

))#y/1以及x/1表示沒有resize

)

3、進行測試

在mmdetection資料夾下,執行:

python demo/demo_test.py

輸出(427,640,3),並在mmdetection/demo資料夾找到demo_result.jpg,即可。

mmdetection 訓練資料

修改數目類別 下面展示一些內聯 片。model dict roi head dict bbox head dict num classes 1 修改檢測類別名稱 修改檢測類別 注意 當目標類別為一類時需要檢測類別為列表形式不是元組將檢測檔案複製到虛擬環境中。若上述檔案發生變換需要重新執行,重新編譯。...

mmdetection 使用技巧篇

在config檔案中新增 log config dict interval 50,hooks dict type textloggerhook dict type tensorboardloggerhook 生成tensorboard 日誌 設定之後,會在work dir目錄下生成乙個tf logs...

mmdetection之dataset類解讀

總結本篇是mmdetection原始碼解讀第二篇,主要講解mmdetection是初始化資料類的。本文以coco資料集為例,當然,原始碼解讀不可能面面俱到,重要的是揣摩設計者的思想以及實現過程。另外,本文先暫時不予介紹dataloader構建過程。通常我們利用pytorch讀取資料集需要構建兩個部分...