yolo augmentation 관련
361
작성한 질문수 27
안녕하세요 선생님, yolo를 학습할 때, augmentation이 적용되는 건가요?
제가 train.py부분 yolo.py부분 등 찾아봐도 augmentation이 적용된 곳을 못찾았습니다.
혹시 augmentation하는 부분이 있다면 위치좀 알려주시면 감사하겠습니다!
답변 1
0
안녕하십니까,
keras yolo3 는 augmentation을 학습 시 적용하지 않았습니다. 아마도 공간 기반 augmentation을 하게되면 bbox 좌표값이 달라질 수 있기 때문인것으로 생각됩니다.
keras yolo3는 augmentation을 하지 않기 때문에 keras의 ImageDataGenerator 객체를 사용하지 않고 직접 Generator 함수를 생성해서 만듭니다.
https://github.com/qqwweee/keras-yolo3/blob/master/train.py의
165 line쯤에 해당 data generator가 있습니다.
| '''data generator for fit_generator''' |
| n = len(annotation_lines) |
| i = 0 |
| while True: |
| image_data = [] |
| box_data = [] |
| for b in range(batch_size): |
| if i==0: |
| np.random.shuffle(annotation_lines) |
| image, box = get_random_data(annotation_lines[i], input_shape, random=True) |
| image_data.append(image) |
| box_data.append(box) |
| i = (i+1) % n |
| image_data = np.array(image_data) |
| box_data = np.array(box_data) |
| y_true = preprocess_true_boxes(box_data, input_shape, anchors, num_classes) |
| yield [image_data, *y_true], np.zeros(batch_size) |
| def data_generator_wrapper(annotation_lines, batch_size, input_shape, anchors, num_classes): |
| n = len(annotation_lines) |
| if n==0 or batch_size<=0: return None |
return data_generator(annotation_lines, batch_size, input_shape, anchors, num_classes)
| def data_generator(annotation_lines, batch_size, input_shape, anchors, num_classes): |
강의 환경설정 질문
0
38
2
Custom Dataset에서의 polygon 정보 관련
0
81
3
cvat.ai 보안 수준이 궁금합니다
0
77
2
캐클 nucleus 챌린지 runpod 실습 코드 에러 질문드립니다.
0
93
3
추론 결과의 Precision(또는 mAP) 평가 방법
0
81
2
mmdetection mask rcnn inferenct 실습 시 runpod 템플릿 관해서 질문드립니다.
0
57
2
runpod에서 google drive 연결 시 오류 발생
0
106
2
로드맵 선택
0
64
1
mmcv
0
55
2
Anchor box의 Positive 처리 위치
0
59
2
해당 강의 runpod 적용 후 에러 제보드립니다
0
83
2
run pod credit 관련 제보
0
93
2
mmdetection 2.x과 3.x 호환 관련 표기
0
75
2
mm_faster_rcnn_train_kitti.ipynb 실행 오류
0
93
3
질문 드립니다.
0
74
3
mm_faster_rcnn_train_coco_bccd 실행 오류 질문드립니다.
0
75
1
강사님께 수정을 제안드리고 싶은 것이 있습니다.
0
91
1
google automl efficientdet 다운로드 및 설치 오류
0
71
1
이상 탐지에 사용할 비전 기술 조언 부탁드립니다.
0
100
2
OpenCV 관련 질문드립니다.
0
68
2
mmcv 설치관련해서 문의드려요
0
326
3
강의 구성 관련해서 질문이 있습니다
1
133
2
모델 변환 성능 질문드립니다.
0
122
1
NMS 로직 문의 드려요
0
111
2