inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

Oxford Pet 데이터로 Train 실습 - Config 설정 및 Train 수행

train_detector()에서 오류가 납니다ㅜㅠ

1730

dlgpqls98

작성한 질문수 8

0

 

 

   train_detector()에서 "AttributeError: 'ConfigDict' object has no attribute 'device'" 이런 오류가 나는데

뭐가 문제인가요..?

 

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

답변 2

4

권 철민

음, 그렇군요.

확인해 보니 mmdetection 패키지에서 내부 변경이 있었나 보군요.

일단 이걸 bug fix로 수정할지 좀 더 지켜봐야 할 것 같습니다.  오류를 수정하려면 config 설정에

cfg.device='cuda'

를 추가하면 됩니다.  아래의 cfg 설정에서 맨 마지막과 같이 추가 하시면 됩니다. 

좋은 정보 감사합니다. 

 

# dataset에 대한 환경 파라미터 수정. 
cfg.dataset_type = 'PetDataset'
cfg.data_root = '/content/data/'

# train, val, test dataset에 대한 type, data_root, ann_file, img_prefix 환경 파라미터 수정. 
cfg.data.train.type = 'PetDataset'
cfg.data.train.data_root = '/content/data/'
cfg.data.train.ann_file = 'train.txt'
cfg.data.train.img_prefix = 'images'

cfg.data.val.type = 'PetDataset'
cfg.data.val.data_root = '/content/data/'
cfg.data.val.ann_file = 'val.txt'
cfg.data.val.img_prefix = 'images'

# class의 갯수 수정. 
cfg.model.roi_head.bbox_head.num_classes = 37
# pretrained 모델
cfg.load_from = 'checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth'

# 학습 weight 파일로 로그를 저장하기 위한 디렉토리로 구글 Drive 설정. 
cfg.work_dir = '/mydrive/pet_work_dir'

# 학습율 변경 환경 파라미터 설정. 
cfg.optimizer.lr = 0.02 / 8
cfg.lr_config.warmup = None
cfg.log_config.interval = 5

cfg.runner.max_epochs = 5

# 평가 metric 설정. 
cfg.evaluation.metric = 'mAP'
# 평가 metric 수행할 epoch interval 설정. 
cfg.evaluation.interval = 5
# 학습 iteration시마다 모델을 저장할 epoch interval 설정. 
cfg.checkpoint_config.interval = 5

# 학습 시 Batch size 설정(단일 GPU 별 Batch size로 설정됨)
cfg.data.samples_per_gpu = 4

cfg.device='cuda'

0

권 철민

안녕하십니까, 

이것만 봐서는 뭐가 문제인지 모르겠군요. 

config 만드실때 device관련한 뭔가 설정을 하신 건가요? 

아님,  맨 처음 오류를 보면  no such file or direction 'mmdetection' 오류가 나는데, mmdetection은 설치가 제대로 된건가요? 

mmdetection이 제대로 설치가 되었는지 강의를 보시고 다시 한번 확인 부탁드리고, 그래도 오류가 발생하면 설정한 config와 수행한 소스코드를 여기에 올려 주십시요. 

감사합니다. 

0

soxn3579

저도 똑같은 오류가 납니다. 수업 도중이라서 바꾼 코드는 없습니다 ㅜㅜ

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

114

2

OpenCV 관련 질문드립니다.

0

88

2

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

0

362

3

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

1

141

2

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

0

129

1