glee1228 / ServerAnalysisModule

0 stars 0 forks source link

object detector class label #3

Open glee1228 opened 4 years ago

glee1228 commented 4 years ago

ROOT_DIR/models/detectors/yolo/data/coco.names

80 Classes label index(0~79) from top to bottom

person bicycle car motorbike aeroplane bus train truck boat traffic light fire hydrant stop sign parking meter bench bird cat dog horse sheep cow elephant bear zebra giraffe backpack umbrella handbag tie suitcase frisbee skis snowboard sports ball kite baseball bat baseball glove skateboard surfboard tennis racket bottle wine glass cup fork knife spoon bowl banana apple sandwich orange broccoli carrot hot dog pizza donut cake chair sofa pottedplant bed diningtable toilet tvmonitor laptop mouse remote keyboard cell phone microwave oven toaster sink refrigerator book clock vase scissors teddy bear hair drier toothbrush

glee1228 commented 4 years ago

검출하고자 하는 클래스 설정은 /ROOT_DIR/SimpleHRNet.py 의 YOLOv3 클래스 객체 생성하는 부분에서 classes 매개변수에 위의 클래스 이름들을 리스트에 추가 삽입하면 됩니다.