inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

Yolo V5로 COCO형태의 BCCD 데이터 Train 실습 - Dataset yaml 적용 후 Train과 Inference 수행

yolov5 강의에서 test.py

4040

kimhy365

작성한 질문수 6

0

선생님, 안녕하세요?
yolov5로 모델을 만들고 예측을 해서 mAP를 계산하려고 합니다.
1. test.py를 이용하려고 하는데, 이상하게 제 yolov5에는 val.py가 있고 test.py가 없습니다. 같은 기능으로 파일명이 변경된 건가요?
2. detect.py로 결과를 파일로 저장해서 그 파일을 GT파일의 값과 비교해서 계산해야 하는 건가요?
좋은 강의에 감사드립니다.

yolov5 머신러닝 배워볼래요? 딥러닝 keras python 컴퓨터-비전 tensorflow

답변 1

1

권 철민

안녕하십니까, 

1. 음, 이게 val.py 로 바뀌었군요. test.py 대신 val.py로만 바꿔서 수행하시면 될 것 같습니다. 

2. detect.py는 하나의 이미지만 detect 합니다. 일반적으로 하나의 파일만 가지고는 mAP를 제공하지는 않습니다.  val.py를 수행하시면 mAP가 나옵니다.  detect.py의 결과를 file로 저장하고자 하신다면  detect.py에 --save-txt라는 option이 있습니다. 

아래와 같이 수행해 보시면, project 디렉토리인 content/data/output 밑에 labels라는 디렉토리가 생기고 여기에BloodImage_0011.txt 라는 파일로 결과가 출력됩니다.  Detect된 object들이 라인별로 출력되면, 맨 앞이 클래스 id입니다. 좌표들은 학습데이터와 동일하게 0~1 사이의 스케일링된 값으로 나오므로 다시 image size를 곱해서 만들어야 합니다.  

!cd /content/yolov5;python detect.py --source /content/bccd/images/test/BloodImage_00011.jpg \
                            --weights /mydrive/ultra_workdir/bccd/weights/best.pt --conf 0.2 \
                            --project=/content/data/output  --save-txt \
                            --name=run_image --exist-ok --line-thickness 2

0

kimhy365

아... 정말 감탄.

이렇게 빨리, 자상하게 답변해 주시니 막혔던 속이 시원해지네요.

감사합니다.

강의 환경설정 질문

0

39

2

Custom Dataset에서의 polygon 정보 관련

0

83

3

cvat.ai 보안 수준이 궁금합니다

0

80

2

캐클 nucleus 챌린지 runpod 실습 코드 에러 질문드립니다.

0

95

3

추론 결과의 Precision(또는 mAP) 평가 방법

0

85

2

mmdetection mask rcnn inferenct 실습 시 runpod 템플릿 관해서 질문드립니다.

0

59

2

runpod에서 google drive 연결 시 오류 발생

0

108

2

로드맵 선택

0

66

1

mmcv

0

57

2

Anchor box의 Positive 처리 위치

0

61

2

해당 강의 runpod 적용 후 에러 제보드립니다

0

85

2

run pod credit 관련 제보

0

97

2

mmdetection 2.x과 3.x 호환 관련 표기

0

77

2

mm_faster_rcnn_train_kitti.ipynb 실행 오류

0

96

3

질문 드립니다.

0

76

3

mm_faster_rcnn_train_coco_bccd 실행 오류 질문드립니다.

0

78

1

강사님께 수정을 제안드리고 싶은 것이 있습니다.

0

93

1

google automl efficientdet 다운로드 및 설치 오류

0

73

1

이상 탐지에 사용할 비전 기술 조언 부탁드립니다.

0

102

2

OpenCV 관련 질문드립니다.

0

70

2

mmcv 설치관련해서 문의드려요

0

329

3

강의 구성 관련해서 질문이 있습니다

1

135

2

모델 변환 성능 질문드립니다.

0

122

1

NMS 로직 문의 드려요

0

112

2