코랩

[코랩] Darknet 설치 및 커스텀 모델 학습하기

IntegerString 2020. 8. 17. 09:25

구글 드라이브와 마운트

colab에 설치되어있는 cuda 버전 확인

cuda 버전에 알맞는 cudnn을 구글 드라이브에 다운로드 후 압축 해제

cudnn 버전 확인

darknet 다운로드

Makefile 변경 (cuda, cudnn, opencv)

darknet 컴파일

컴파일한 darknet을 구글 드라이브로 복사

미리 학습된 가중치 파일(yolov4.conv.137)을 구글 드라이브에 다운로드

darknet 권한 변경 및 실행 테스트

./darknet: error while loading shared libraries: libcudnn.so.8: cannot open shared object file: No such file or directory

만약 위와 같은 에러가 나온다면 ./darknet 앞에 LD_LIBRARY_PATH=/usr/local/cuda/lib64/ 를 추가

학습 실행

윈도우 환경과는 다르게 GUI를 지원하지 않으므로 -dont_show 옵션이 필요함

파일 목록

- 유용한 JS 코드

https://hanryang1125.tistory.com/31

 

코랩 런타임 연결 유지 및 출력 삭제 (Colab)

개발자 도구(F12)를 열어 콘솔에 입력 - 연결 유지 (코랩의 최대 세션 유지 시간은 12시간) function ClickConnect(){ console.log("1분마다 코랩 연결 끊김 방지"); document.querySelector("#top-toolbar > cola..

hanryang1125.tistory.com

참고자료: https://colab.research.google.com/drive/1lTGZsfMaGUpBG4inDIQwIJVW476ibXk_

 

Tutorial DarknetToColab.ipynb

Colaboratory notebook

colab.research.google.com

참고자료 2: colab.research.google.com/drive/12QusaaRj_lUwCGDvQNfICpa7kA7_a2dE

 

Google Colaboratory

 

colab.research.google.com