프로그래밍 언어

[Python] 구글 Colab을 활용한 YOLO 모델 학습

구글 Colab의 경우 클라우드 기반의 주피터 노트북 개발환경으로 별도의 파이썬 설치가 필요 없으며 데이터 분석에 용이한 패키지들이 기본적으로 내장되어 있다. 또한 구글 드라이브 마운트 시 자신의 구글 드라이브, 깃과 연동하여 파일을 클라우드 형태의 저장소에 보관하며 협업 및 관리할 수 있다. 가장 큰 장점이라면 자신이 저사양 노트북이나 PC 환경이더라도 공간 제약을 받지 않으며 Google Colab에서 제공하는 GPU를 제한적이지만 무료로 사용 가능하다는 것이다. 우선 YOLO 모델을 Colab 가상 환경에서 학습하기 위해 Colab에 접속한다. https://colab.google/ colab.google Colab is a hosted Jupyter Notebook service that requ..

[Python] YOLO 모델 학습을 위한 오픈 데이터 셋 준비 (2)

이전 포스팅 [Python] YOLO 모델 학습을 위한 오픈 데이터 셋 준비 (1) [Python] YOLO 모델 학습을 위한 오픈 데이터 셋 준비 (1) =============== 목표 =============== YOLO 모델 학습을 위한 오픈 데이터셋 준비 ================================== 1. roboflow를 통해 배포되어 있는 오픈 이미지 데이터 셋 다운 2. 구글 OpenImageDataset과 Google BigQuery coding-gym.tistory.com 《 OIDv4_ToolKit 활용 》 OIDv4_ToolKit GitHub - EscVM/OIDv4_ToolKit: Download and visualize single or multiple class..

[Python] YOLO 모델 학습을 위한 오픈 데이터 셋 준비 (1)

=============== 목표 =============== YOLO 모델 학습을 위한 오픈 데이터셋 준비 ================================== 1. roboflow를 통해 배포되어 있는 오픈 이미지 데이터 셋 다운 2. 구글 OpenImageDataset과 Google BigQuery를 활용해 원하는 일부 데이터 셋 추출 3. OIDv4_ToolKit 활용 《 roboflow 활용 》 roboflow 공식 홈페이지에서 제공하는 데이터셋을 다운로드 roboflow Computer Vision Datasets Download free, open source datasets for computer vision machine learning models in a variety of ..

[Python] YOLOv5를 활용한 객체 인식(Object Detection) (2)

이전 포스팅에서 이미지 라벨링 작업을 완료 했다면 라벨링 된 데이터를 YOLO 형식으로 전환 해주도록 하겠습니다. 《 데이터 셋 형식 변환 》 현재 CVAT을 통해 다운받은 데이터는 COCO형식의 json 파일입니다. 이를 Yolov5에서 사용하기 위해 convert2Yolo를 활용하도록 하겠습니다. covert2Yolo 패키지와 yolov5의 경우 아래의 git 페이지에서 zip 형식을 통해 다운받아도 되고 콘솔 혹은 터미널을 통해 다운로드 받을 수도 있습니다. [.zip 형식으로 직접 다운] yolov5 깃허브 페이지 GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite YOLOv5 🚀 in PyTorch > ONNX > ..

[Python] YOLOv5를 활용한 객체 인식(Object Detection) (1)

============= 목표 ============= YOLOv5 모델을 통한 선박 객체 인식 ============================== 1. 데이터 셋 준비하기 2. 데이터 셋 형식 변환 3. 데이터 셋을 통한 모델 학습 4. 훈련 결과를 통한 객체 인식 검증 《 데이터 셋 준비하기 》 모델 학습을 위한 데이터 셋 준비가 필수적이다. YOLO를 학습하기 위한 데이터 셋 형태는 아래와 같다 ''' 학습을 위한 이미지 파일 각 이미지와 1:1로 매핑되는 YOLO 형식의 annotaion 텍스트 파일 ''' YOLOv5의 어노테이션 형식은 .txt의 텍스트파일으로 해당 텍스트 파일들은 각 이미지와 매핑되어 이미지의 바운딩 박스에 대한 설명을 위한 자료로 활용된다. 예를 들어 아래와 같은 이미지..