Python影像辨識筆記(二十):detectron2

detectron2是Facebook提供的Mask R-CNN的Python API,基於同樣屬於Facebook研發的PyTorch框架,使用起來相當方便。讓使用者可以在很短的時間內就完成Semantic Segmentation(語意分割)的任務。

Model Zoo的地方提供可各種不同的Model供使用者應用,該文為官方部落格的介紹

安裝

https://github.com/facebookresearch/detectron2/blob/master/INSTALL.md

git clone https://github.com/facebookresearch/detectron2.git
python -m pip install -e detectron2 #不用cd到detectron2資料夾
or python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu92/torch1.6/index.html

使用

Colab連結中的ipynb主要進行套件引入、模型下載、推論、訓練、推論圖片顯示......等功能。# 注意事項
在aiForge上如果要顯示圖片的話,Colab中提供的cv2_imshow(im)會有錯誤產生,必須改用下列方式在Jupyter中的cell顯示圖片
from IPython.display import display
from PIL import Image
import cv2
im = cv2.imread("./input.jpg")
im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB) # Converting BGR to RGB
display(Image.fromarray(im))

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store