Ground truth bounding box를 이미지와 함께 시각화하는 방법을 알고싶습니다.
974
작성한 질문수 6
Custom dataset으로 YOLO 학습을 배우는 중인 학생입니다.
그런데 작업 중 Ground truth bounding box를 이미지에 같이 출력하는 방법을 도저히 모르겠어서 질문 남깁니다.
지금은 아래처럼 score가 있는 predicted image만 detect.py 파일로 만들 수 있습니다.
여기서 스코어 없이 Label만 annotation 파일에 기반하여 달린 이미지를 얻고싶습니다.
Wandb 학습 중 만들어지는 파일은 아래처럼 알아볼 수 없어서 따로 요청드립니다.
현재 작업중인 파일은 COCO 포맷의 커스텀 데이터셋을 강사님이 제공해주신 COCO2YOLO 코드로 변환했기 때문에, 디렉토리는 강의와 똑같습니다.
예시: /content/custom/images/train/*.png - /content/custom/labels/train/*.txt
Ultralytics YOLO 패키지 내부의 파일로 해결하기 힘들다면 다른 우회 방법이라도 알고싶습니다.
항상 좋은 강의 잘 듣고 있습니다.
감사합니다.
답변 1
0
안녕하십니까,
질문하신 내용을 제가 정확히 파악했는지 모르겠습니다. 예측 할때가 아니라 학습 iteration시에 검증 데이터로 임시 예측한 이미지 파일을 얻고 싶으신건가요?
만약 그렇다면 저도 wandb 사용하는 방법외에는 잘 모르겠습니다.
일단 지금 어지럽게 annotation이 나오는 것은 학습이 충분히 되지 않아서 그런것 같습니다. no-demage가 너무 많이 나오는 군요. 전체 오브젝트에서 no-demage가 차지하는 부분이 상대적으로 많은데 거기에 학습까지 충분히 되지 않아서 그런걸로 생각됩니다.
좀 더 iteration을 많이 주면서 중간 중간에 wandb에서 나오는 이미지를 확인해 보시면 어떨까 싶습니다.
감사합니다.
MMDetection 버전 이슈
0
59
2
강의 환경설정 질문
0
66
2
Custom Dataset에서의 polygon 정보 관련
0
114
3
cvat.ai 보안 수준이 궁금합니다
0
101
2
캐클 nucleus 챌린지 runpod 실습 코드 에러 질문드립니다.
0
120
3
추론 결과의 Precision(또는 mAP) 평가 방법
0
98
2
mmdetection mask rcnn inferenct 실습 시 runpod 템플릿 관해서 질문드립니다.
0
71
2
runpod에서 google drive 연결 시 오류 발생
0
128
2
로드맵 선택
0
74
1
mmcv
0
66
2
Anchor box의 Positive 처리 위치
0
71
2
해당 강의 runpod 적용 후 에러 제보드립니다
0
97
2
run pod credit 관련 제보
0
129
2
mmdetection 2.x과 3.x 호환 관련 표기
0
89
2
mm_faster_rcnn_train_kitti.ipynb 실행 오류
0
115
3
질문 드립니다.
0
89
3
mm_faster_rcnn_train_coco_bccd 실행 오류 질문드립니다.
0
90
1
강사님께 수정을 제안드리고 싶은 것이 있습니다.
0
103
1
google automl efficientdet 다운로드 및 설치 오류
0
87
1
이상 탐지에 사용할 비전 기술 조언 부탁드립니다.
0
113
2
OpenCV 관련 질문드립니다.
0
88
2
mmcv 설치관련해서 문의드려요
0
361
3
강의 구성 관련해서 질문이 있습니다
1
141
2
모델 변환 성능 질문드립니다.
0
129
1