'커스텀' 태그의 글 목록
본문 바로가기

커스텀3

YOLOR custom tutorial (YOLOR 커스텀 데이터셋 학습) - 2편(Train) 설치(Installation) # 원하는 디렉토리로 이동해서 YOLOR git clone! git clone https://github.com/roboflow-ai/yolor cd yolor git reset --hard eb3ef0b7472413d6740f5cde39beb1a2f5b8b5d1 # Requirement 설치 pip install -r requirements.txt ※ Requirement 설치 시 ERROR: torch has an invalid wheel, .dist-info directory not found 오류가 발생한다면, PyTorch 홈페이지에 가서 버젼에 맞게 다운받는다. conda install pytorch==1.7.0 torchvision==0.8.1 torchaudi.. 2021. 7. 20.
YOLOR custom tutorial (YOLOR 커스텀 데이터셋 학습) - 1편(데이터셋 제작) roboflow활용해서 데이터셋 구축하고, 추후 커스텀 annotation tool을 활용해서 학습하는 방법을 알아보겠다. 왜냐면, roboflow는 1000장까지밖에 무료로 데이터셋을 제공하기 때문. 기존에 yolo 시리즈 데이터셋을 제작하는 annotation툴이 있다면 그대로 사용해도 무관하다. 목차 1. 데이터셋 만들기 2. 기본환경설치 3. 트레이닝 4. 테스트 1. 데이터셋 만들기(https://roboflow.ai/) ROBOFLOW 접속 - 회원가입 - create project - Project Name(프로젝트이름), Type(오브젝트디텍션), Annotation Group(클래스이름 입력) 라벨링 할 이미지를 드래그앤드랍으로 가져온다음 Finish Uploading 버튼 클릭.(이미지.. 2021. 7. 20.
Custom Detectron2 Training!! (Person Detection)-챕터 2 모델 훈련시키기(구글 colab) 목차 EP1. 데이터셋 만들기 EP2. 모델 훈련시키기 EP3. 정확도 확인하기 EP4. 예측하기 2. 모델 훈련시키기 구글드라이브 - 새폴더 생성(ex. detectron2) 마우스우클릭 - 더보기 - Google Colaboratory 클릭 런타임(화면상단) - 런타임유형변경 - None->GPU로 변경 왼쪽 폴더 - 드라이브마운트(중앙폴더) - Google Drive에 연결 클릭 폴더를 하나씩 눌러보면 drive/My Drive/Coding/detectron2 폴더가 보인다. 설치시작! 폴더 우클릭 - 경로복사 - 코드 첫 줄에 %cd /content/drive/My Drive/Coding/detectron2 입력 %cd /content/drive/My Drive/Coding/detectron2 t.. 2020. 10. 12.