inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

tiny kitti 데이터로 MMDetection Train 실습 - Train 실행 후 이미지 Inference

train_detection을 하는데 오류가 발생합니다

238

Yejoon Cho

작성한 질문수 5

0

코랩이 아닌, 제 로컬에서 학습을 하고있습니다. Kitty_tiny 커스텀 데이터셋을 만드는것까진 되는데, train_detection을 하는과정에서 오류가 뜨네요.
Study/Kitty_Dataset.ipynb at master · GreedyNote/Study (github.com)
전체 오류코드는 링크를 올리겠습니다.
RAM이 16GB이긴한데 램용량이 부족해서 그런것같기도 합니다. 혹시 램때문이라면 MMdetection에서 Batch 사이즈를 조정할 수 있는 방법이 있나요?

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

답변 1

0

권 철민

안녕하십니까, 

 batch size는 config로 잡으시면 되는데

cfg.data.samples_per_gpu 로 설정하면 됩니다. 현재 samples_per_gpu는 2로 설정되어 있습니다. 1로 설정하고 함 테스트 해보시지요.  그리고 GPU Memory에 다른 것이 할당되어 있는지도 확인하시고 있으면 클리어 해주시지요. 

이걸로도 문제가 해결되지 않으면, 뭔가 mmdetection과 pytorch가 잘 안맞는 것 같습니다.

가급적이면 windows보다는 vm으로 linux상에서 설치를 하는것이 좋을 것 같습니다.  그리고 windows 설치시에 아래 방식대로 설치 되었는지 확인 부탁드립니다. 

https://mmcv.readthedocs.io/en/latest/get_started/build.html

MMDetection 버전 이슈

0

52

2

강의 환경설정 질문

0

61

2

Custom Dataset에서의 polygon 정보 관련

0

113

3

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

0

98

2

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

0

120

3

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

0

94

2

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

0

68

2

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

0

126

2

로드맵 선택

0

74

1

mmcv

0

65

2

Anchor box의 Positive 처리 위치

0

71

2

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

0

95

2

run pod credit 관련 제보

0

125

2

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

0

89

2

mm_faster_rcnn_train_kitti.ipynb 실행 오류

0

114

3

질문 드립니다.

0

89

3

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

0

89

1

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

0

102

1

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

0

87

1

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

0

111

2

OpenCV 관련 질문드립니다.

0

87

2

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

0

357

3

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

1

141

2

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

0

126

1