MMDetection & Detectron2
# 모듈 import from mmcv import Config from mmdet.datasets import build_dataset from mmdet.models import build_detector from mmdet.apis import train_detector from mmdet.datasets import (build_dataloader, build_dataset, replace_ImageToTensor) classes = ("General trash", "Paper", "Paper pack", "Metal", "Glass", "Plastic", "Styrofoam", "Plastic bag", "Battery", "Clothing") # config file 들고오기 cfg = Con..