Detection 3

Detectron2 - Training 응용

detectron2 official example : https://github.com/facebookresearch/detectron2/tree/main/projects 간략한 학습 코드 : https://github.com/nhm0819/detectron2_training 이번에는 detectron2 튜토리얼에 있는 과정 말고, 조금 더 customize 하는 방법에 대해 소개하려고 합니다. detectron2 tutorial에서 학습하는 코드를 보면 아래와 같이 되어있습니다. from detectron2.config import get_cfg from detectron2.engine import DefaultTrainer cfg = get_cfg() # default configure cfg.DAT..

Detection 2022.02.08

Detectron2 - 설치, Training 기초

object detection을 위한 api로는 대표적인게 Detectron, MMDetection, YOLOv5, 이 3가지 인 것 같습니다. 사실 제가 그냥 아는 api가 3개 밖에 없네요..ㅋㅋ 요즘 보면 Detectron보단 MMDetection이 훨씬 더 많은 모델들을 지원하고, 또 자주 업데이트 하는 것 같네요. 뭐 여하튼 본 글은 Detectron2에 대한 설명 글입니다. 0. 설치 linux, macOS 는 공식 문서에 나와있는 절차대로 따라하시면 쉽게 설치가 됩니다. (https://detectron2.readthedocs.io/en/latest/tutorials/install.html) 문제는 windows인데요, windows는 공식적으로 지원하고 있진 않습니다. windows는 아래..

Detection 2022.02.07

Mask RCNN (Faster RCNN + segmentation)

mask rcnn의 논문 리뷰는 인터넷에 수도 없이 많기 때문에, 논문 리뷰보다는 코드를 보면서 리뷰를 하겠습니다. 논문에 대한 설명은 다른 곳으로.. 이 글에서 사용할 코드 : nhm0819/mask_rcnn_prac (github.com) - pytorch tutorial이며 torchvision에 있는 mask rcnn 모델을 기반으로 합니다. (readme에 올려놓은 dataset 받아서 debugging 해보시면 됩니다.) * 본 글에 달아놓은 코드들은 제가 상당히 많은 부분을 삭제한 pseudo code 입니다. 요약 먼저 mask rcnn은 faster rcnn에 segmentation 기능만 추가되었다고 봐도 무방할 정도로 거의 비슷합니다. 전체적인 흐름도는 다음과 같습니다. - 이미지가 ..

Detection 2022.02.06