YOLOv5 검출 결과 Crop

실행 환경: Google Colab

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

import torch
import os

# Model
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')  # or yolov5n - yolov5x6, custom


file_list = os.listdir('../test')

for i in file_list:
    img='../test/'+i
    results = model(img)
    crops=results.crop(save=True)

결과

결과 local에 저장

#zip [압축파일명.zip] [압축할 파일/폴더명]
!zip -r ../file.zip ../result

from google.colab import files
files.download("/content/file.zip")