깃허브에 있는 딥러닝 모델 가져오기(깃허브 오픈소스 사용법)
2975
작성자 없음
작성한 질문수 0
안녕하세요 아직 많이 듣지는 않았지만 ㅎ 유익한 강의 정말 잘 듣고 있습니다.!
다름이 아니라 최근 초해상화(Super resolution)에 관심이 다양한 딥러닝 모델들을 찾아보았습니다.
아직 논문을 보고 코드를 구현하기에는 많이 부족해서 다른 사람들이 깃허브에 업로드한 딥러닝 모델들을 적용해보고 싶은데 활용방법에 익숙치 않습니다.
예를들어 SRCNN모델을 적용해보고싶어 https://github.com/TheStarkor/SRCNN-tensorflow2 이 깃허브를 활용하고 싶을때 어떻게 자신만의 데이터를 학습시키고 output 데이터를 도출할 수 있는지 잘 감이 잡히지 않아 질문드립니다.
답변 1
0
안녕하십니까,
음, 저도 이걸 구동해 보진 않았지만 main.py를 해석해 봤습니다. 너무 길어서 여기서 적어서 설명드릴수는 없을것 같습니다. 스스로 main.py를 함 보셔야 할 것 같습니다.
개요적으로 말씀드리면 아래 구성에서
DATA_DIR = "../src/"
FILE_PATH = "./models/srcnn_div2k.hdf5"
TRAIN_PATH = "DIV2K_train_HR"
TEST_PATH = "DIV2K_valid_HR"
N_TRAIN_DATA = args.N_TRAIN_DATA
N_TEST_DATA = args.N_TEST_DATA
BATCH_SIZE = args.BATCH_SIZE
EPOCHS = args.EPOCHS
train_data_generator() 함수 내부를 보시면 ImageDataGenerator.flow_from_directory() 구조로 되어 있습니다. 학습 데이터를 flow_from_directory()에서 사용가능할 수 있도록 DATA_DIR = "../src/" 디렉토리 구조에 맞게 옮겨 주셔야 학습이 될 것 같습니다.
MMDetection 버전 이슈
0
51
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
355
3
강의 구성 관련해서 질문이 있습니다
1
141
2
모델 변환 성능 질문드립니다.
0
125
1