설치

python=3.8, pytorch=2.2.2, torchvision=0.17로 설치

pip3 install torch torchvision==0.17 pyyaml==5.1

detectron2 설치

가상환경 활성화 후, detectron2 폴더가 있는 위치에서 해당 코드로 설치
python -m pip install -e detectron2

detectron2 class ID

다음 코드로 확인 가능하다

from detectron2.data import MetadataCatalog

# COCO 데이터셋의 메타데이터를 가져옴
coco_metadata = MetadataCatalog.get("coco_2017_train")

# 클래스 이름과 ID 출력
for idx, name in enumerate(coco_metadata.thing_classes):
    print(f"ID: {idx}, Class: {name}")

ID: 0, Class: person
ID: 1, Class: bicycle
ID: 2, Class: car
ID: 3, Class: motorcycle
ID: 4, Class: airplane
ID: 5, Class: bus
ID: 6, Class: train
ID: 7, Class: truck
ID: 8, Class: boat
ID: 9, Class: traffic light
ID: 10, Class: fire hydrant
ID: 11, Class: stop sign
ID: 12, Class: parking meter
ID: 13, Class: bench
ID: 14, Class: bird
ID: 15, Class: cat
ID: 16, Class: dog
ID: 17, Class: horse
ID: 18, Class: sheep
ID: 19, Class: cow
ID: 20, Class: elephant
ID: 21, Class: bear
ID: 22, Class: zebra
ID: 23, Class: giraffe
ID: 24, Class: backpack
ID: 25, Class: umbrella
ID: 26, Class: handbag
ID: 27, Class: tie
ID: 28, Class: suitcase
ID: 29, Class: frisbee
ID: 30, Class: skis
ID: 31, Class: snowboard
ID: 32, Class: sports ball
ID: 33, Class: kite
ID: 34, Class: baseball bat
ID: 35, Class: baseball glove
ID: 36, Class: skateboard
ID: 37, Class: surfboard
ID: 38, Class: tennis racket
ID: 39, Class: bottle
ID: 40, Class: wine glass
ID: 41, Class: cup
ID: 42, Class: fork
ID: 43, Class: knife
ID: 44, Class: spoon
ID: 45, Class: bowl
ID: 46, Class: banana
ID: 47, Class: apple
ID: 48, Class: sandwich
ID: 49, Class: orange
ID: 50, Class: broccoli
ID: 51, Class: carrot
ID: 52, Class: hot dog
ID: 53, Class: pizza
ID: 54, Class: donut
ID: 55, Class: cake
ID: 56, Class: chair
ID: 57, Class: couch
ID: 58, Class: potted plant
ID: 59, Class: bed
ID: 60, Class: dining table
ID: 61, Class: toilet
ID: 62, Class: tv
ID: 63, Class: laptop
ID: 64, Class: mouse
ID: 65, Class: remote
ID: 66, Class: keyboard
ID: 67, Class: cell phone
ID: 68, Class: microwave
ID: 69, Class: oven
ID: 70, Class: toaster
ID: 71, Class: sink
ID: 72, Class: refrigerator
ID: 73, Class: book
ID: 74, Class: clock
ID: 75, Class: vase
ID: 76, Class: scissors
ID: 77, Class: teddy bear
ID: 78, Class: hair drier
ID: 79, Class: toothbrush