CLASSES 관련 질문입니다.

22.03.18 23:42 작성 조회수 96

강의 잘 듣고 있습니다. 감사드립니다.

CLASSES = ('Car', 'Truck', 'Pedestrian', 'Cyclist')

데이터셋 클래스를 정의해줄 때 반드시 CLASSES라고 변수명을 정해야 한다고 하셨는데요, 왜 그런 건지 궁금합니다. CLASSES가 어디랑 연동이 되어 있길래 그렇게 해야 하는 것인가요?

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

tiny kitti 데이터로 MMDetection Train 실습 - CustomDataset 만들기 01

강의실 바로가기

답변 1

답변을 작성해보세요.

권 철민

2022.03.19

안녕하십니까,

KittyDataset 클래스는 CustomDataset 클래스를 상속받습니다.

이 CustomDataset 클래스가 MMDetection의 주요 모듈에 입력 데이터를 전달하는데 여기서 CLASSES가 정의 되어 있습니다. 그래서 KittyDataset은 이 CLASSES를 오버라이딩 해서 적용합니다.

감사합니다.