BEST
인공지능

/

컴퓨터 비전

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

본 강의는 Object Detection과 Segmentation에 대한 깊이 있는 이론 설명과 현업에서 바로 사용될 수 있는 수준의 실습 예제를 통해 여러분을 현장에서 필요한 딥러닝 기반의 컴퓨터 비전 전문가로 발돋움시켜 드릴 것입니다.

(4.9) 수강평 138개

수강생 3,578명

Thumbnail

mmdetection을 이용하여 train_detector 수행 시 오류 발생 해결 방안 공지

안녕하십니까, 

 

mmdetection 패키지의 최근 변경으로 인하여 train_detector(model, datasets, cfg, distributed=False, validate=True) 수행 시 'ConfigDict' object has no attribute 'device' 와 같은 오류가 발생하고 있습니다. 

 

해당 이슈는 https://github.com/open-mmlab/mmdetection/issues/7901 에서 확인하실 수 있습니다. 

 

해결 방법은 config 설정에 아래와 같이 cfg.device='cuda'  를 추가해 주시면 됩니다. 

 

가령 예를 들어 실습 코드에 있는 cfg 설정에서 아래와 같이 맨 마지막에 cfg.device='cuda'를 추가해 주시면 됩니다. 

 

해당 내용이 Bug인지, 아님 의도적인 것인지 확실하지 않으므로 일단은 github에 있는 실습 코드는 수정하지 않겠습니다. 

 

1주일 정도 지켜본 뒤에 문제가 여전히 발생할 경우 github에 있는 실습 코드를 수정후 공지 드리겠습니다. 

 

감사합니다. 

 

# 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'

 

..... 그밖의 config 설정

 

# 'ConfigDict' object has no attribute 'device' 오류 방지를 위한 설정. 

cfg.device='cuda'

댓글