安裝mmdetection並測試demo

2021-10-02 10:45:55 字數 1380 閱讀 1042

一、安裝mmdetection

(0)ubuntu已有環境

anaconda;cuda–v10.0.130;cudnn–7.4.2

(1)新建conda環境並啟用

conda create --name open-mmlab python=3.6

conda activate open-mmlab

具體也可參考之前的部落格:linux系統下建立anaconda新環境及問題解決

(2)安裝相應版本的pytorch和torchvision

進入查詢適合自己環境的安裝命令

例如,我安裝的pytorch1.1.0

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch
(3)轉殖mmdetection
git clone https:/

(4)安裝mmdetection
python setup.py develop
其他依賴庫會自動安裝,如果還提示缺少什麼庫,根據情況自行安裝即可。

二、測試乙個demo

from mmdet.apis import init_detector, inference_detector, show_result

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)

# 測試一張

#這裡將結果進行儲存,其餘引數為預設值

結果圖:

訓練voc格式的資料集詳見後續部落格:mmdetection訓練pascal_voc格式的資料集

mmdetection學習筆記(一) 安裝

配置要求 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 highe...

mmdetection安裝與環境配置

最近自己真的安裝了太多的計算機視覺環境了 越來越會找bug了,是不是很棒,hh 這次趁熱打鐵總結一下mmdetection的安裝步驟。涉及到 cuda,python,pytorch,mmcv,gcc 去官網上找到對應cuda和python的版本 conda install pytorch 1.5.0...

ubuntu系統安裝並使用svn 親測有效

第一步 安裝svn 本人喜歡在root許可權下操作我一般都是先su root 然後輸入密碼進入root許可權操作 第一步 副 apt get install subversion,然後一鍵式安裝 第二步 進入工作目錄svn checkout 工作目錄路徑 https 埠號 svn 第三步 會有個 r...