프로젝트ㅣ- 시니어 건강관리 음식인식(2) 머신러닝

머신러닝 yolov5모델 

 

from google.colab import drive

 

drive.mount('/content/drive')

드라이브마운트 실행

 

!git clone https://github.com/ultralytics/yolov5.git

%cd /content/drive/MyDrive/projectl/yolov5

 !pip install -r requirements.txt

yolov5 깃에서 다운로드 후 

yolov5가 있는 모델의 경로를 찾아 requirements.txt  를 읽을수 있는 파일 설치

 

import matplotlib.pyplot as plt

plt.rcParams['font.family'] = 'DejaVu Sans'  # 'DejaVu Sans' 부분을 존재하는 폰트로 변경

!apt-get update

!apt-get install -y fonts-liberation

!python train.py --img 960 --batch 16 --epochs 2 --data yolov5data.yaml --weights yolov5s.pt

바로 위코드에서 오류

 

 

Transferred 343/349 items from yolov5s.pt AMP: checks passed ✅ optimizer: SGD(lr=0.01) with parameter groups 57 weight(decay=0.0), 60 weight(decay=0.0005), 60 bias albumentations: Blur(p=0.01, blur_limit=(3, 7)), MedianBlur(p=0.01, blur_limit=(3, 7)), ToGray(p=0.01), CLAHE(p=0.01, clip_limit=(1, 4.0), tile_grid_size=(8, 8)) train: Scanning /content/drive/MyDrive/projectl/dataset/train (1)/images (1).cache... 595 images, 856 backgrounds, 0 corrupt: 100% 1451/1451 [00:00<?, ?it/s] train: WARNING ⚠️ /content/drive/MyDrive/projectl/dataset/train (1)/labels (1)/01_012_01012003_160864648298654_1 (1).jpg: corrupt JPEG restored and saved train: WARNING ⚠️ /content/drive/MyDrive/projectl/dataset/train (1)/labels (1)/01_012_01012003_160890825214430_1 (1).jpg: corrupt JPEG restored and saved train: WARNING ⚠️ /content/drive/MyDrive/projectl/dataset/train (1)/labels (1)/01_012_01012003_160901390416038_1 (1).jpg: corrupt JPEG restored and saved train: WARNING ⚠️ /content/drive/MyDrive/projectl/dataset/train (1)/labels (1)/01_012_01012004_161017772841158_0 (1).jpg: corrupt JPEG restored and save

 

사진하나하나마다 이런 워닝메시지가 뜸 

"corrupt JPEG restored and saved"
이는 손상된 JPEG 파일이 복구되고 저장되었다는 의미로 손상된 파일을 복구해서 저장하였다는 의미 

근데 거의 손상된 파일이라 문제가 있는것같음 . 

 

train 폴더에는 이미지파일, 이미지의 라벨파일

val 폴더에도 이미지파일, 이미지의 라벨파일