• 카테고리

    질문 & 답변

  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

pre_trained 질문

22.02.10 14:17 작성 조회수 160

0

coco dataset에 pretrained된 yolo 모델이 있을때,
coco dataset에 포함되지 않은 custom class를 추가해서 넣고 싶은데 안되나요?
 
쉽게 말씀드리면 coco dataset에 pretrained 된 yolo class에 + custom class를 해서
모델이 기존 class도 분류하면서 custom class도 분류하게 끔하는 방법이요
course-thumbnail

차량 번호판 인식 프로젝트와 TensorFlow로 배우는 딥러닝 영상인식 올인원

답변 1

답변을 작성해보세요.

0

AISchool님의 프로필

AISchool

지식공유자

2022.02.20

안녕하세요~. 반갑습니다.

coco dataset에 포함되지 않은 custom class와 coco dataset에 있는 class를 같이 예측하고자 할때는

coco dataset + custom class로 구성된 데이터셋을 만들어서 다시 학습시켜주어야만 합니다.

아직까지의 기술력으로는 기존의 coco dataset에 학습된 데이터셋에 custom class만을 따로 추가할 수는 없습니다ㅠ

좋은 하루 되세요~.

감사합니다.

 

 

조용현님의 프로필

조용현

질문자

2022.02.22

"coco dataset + custom class로 구성된 데이터셋을 만들어서 다시 학습시켜주어야만 합니다."

라고 하셨는데 coco dataset + custom dataset한 후 coco class + custom class 라벨로 학습시키는 걸로 이해했어요.

 

근데 custom dataset에 coco class가 포함되어 있다면

custom dataset에 coco class도 꼭 labelling 작업을 해줘야 하나요?

custom dataset에 custom class만 ground truth로 만들고 coco dataset + custom class 하면 안되나요? 

 

AISchool님의 프로필

AISchool

지식공유자

2022.02.28

안녕하세요~. 원칙적으로는 custom dataset에도 coco class를 labelling 해주는 것이 맞습니다.

극단적인 예를 들어서 coco dataset에 보라색 자동차 이미지(=car class)가 1개도 없는데, custom dataset에는 보라색 자동차 이미지가 많다면 custom dataset에 car class labelling을 안해주면 트레이닝 데이터셋에 보라색 자동차 이미지가 많음에도 불구하고 보라색 자동차를 잘 검출하지 못하는 검출기로 학습될 것입니다.

다만 coco dataset만으로도 coco class에 대한 학습이 잘되어서 문제가 없을수도 있습니다. 상황마다 다르다고 생각해주시면 될 것 같습니다.

감사합니다.

이 글과 비슷한 Q&A