인프런 커뮤니티 질문&답변
작성한 질문수
RCNN Tensorflow Object detection 에러 및 모델 사용법 질문입니다
해결된 질문
작성
·
333
0
1. 안녕하세요. 저번에 개인 PC에 환경설정을 해도 된다고 하셔서, 저는 개인 PC로 실습을 계속 하고 있다가
RCNN 계열에서 Tensorflow 실습에서 다음과 같이 CUDA 툴킷이 일치하지 않는다고 나와있는데
반드시 10.0 버전으로 설치되어 있어야, 프로그램이 동작을 할까요? 11.1로 받게되면 에러가 발생하는가에 대해서 궁금증이 있습니다
2. 전이훈련으로 제 데이터셋을 훈련하고 싶은데, 이 때 다운로드 받으려는 모델에 대해서 주의사항이 있을까요? 훈련이 된(가중치가 있는) 모델을 무조건 다 받는게 아니라 'pre-trained'라고 명시된 것만 다운로드 받아야 하나요? 차이가 있는지 궁금합니다. 마지막 output layer가 잘린 것으로 받아야 하는건지 이 점이 궁금합니다
감사합니다
답변 2
1
안녕하십니까,
1. Tensorflow 1.x 버전은 Cuda 10.0 이 있어야 합니다. 11과는 호환되지 않습니다.
2. 음, 사실 다운로드 받으려는 모델 사이트가 어디이고 어떤 형태인지 몰라서 답변을 어떻게 드려야 할지 모르겠습니다. pre-trained라고 명시 되어 있는것은 이미 coco등의 데이터 세트로 학습이 완료된 모델로 생각됩니다. pre-trained가 아니라면 모델은 동일하지만 random 값으로 weight가 설정되어 있는 것 같습니다. 마지막 output을 자르시려면 다운로드 받으신 후에 잘라도 무방합니다.
3. coco로 훈련된 모델을 다시 PASCAL VOC로 훈련해도 상관은 없습니다만, 굳이 많은 데이터 세트로 학습된 모델을 작은 데이터 세트로 재 학습할 필요는 없어 보입니다.
감사합니다.
0
