How to crop-save more than 1 image file?

[Raymon360]

Search before asking

• I have searched the YOLOv5 issues and discussions and found no similar questions.

Question

How to crop-save more than 1 image file in multiple folders with size according to trained detection model bounding box?

Additional

No response

[barney2074]

In detect.py- you have the --save-crop option

[glenn-jocher]

@Raymon360 👋 Hello! Thanks for asking about cropping results with YOLOv5 🚀. Cropping bounding box detections can be useful for training classification models on box contents for example. This feature was added in PR #2827. You can crop detections using either detect.py or YOLOv5 PyTorch Hub:

detect.py

Crops will be saved under runs/detect/exp/crops, with a directory for each class detected.

python detect.py --save-crop

YOLOv5 PyTorch Hub

Crops will be saved under runs/detect/exp/crops if save=True, and also returned as a dictionary with crops as numpy arrays.

import torch

# Model
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')  # or yolov5m, yolov5l, yolov5x, custom

# Images
img = 'https://ultralytics.com/images/zidane.jpg'  # or file, Path, PIL, OpenCV, numpy, list

# Inference
results = model(img)

# Results
crops = results.crop(save=True)  # or .show(), .save(), .print(), .pandas(), etc.

Good luck 🍀 and let us know if you have any other questions!

[Raymon360]

Thanks for the response, but can cut save detection be used to detect multiple image files at once?

e.g. 10 images from each subfolder

[barney2074]

@Raymon360
you might need to provide more info- sample, screenshot etc, or at least describe what you are doing
It sounds like you are feeding in a single image to detect.py. If you input multiple images (a folder, or glob path)- then you'll get multiple cropped outputs

[Raymon360]

you might need to provide more info- sample, screenshot etc, or at least describe what you are doing
It sounds like you are feeding in a single image to detect.py. If you input multiple images (a folder, or glob path)- then you'll get multiple cropped outputs

can you provide a code example of what you are describing?

[barney2074]

Yolov5/Ultralytics documentation is pretty good- I'd start by reading it

[Raymon360]

oh I just got it, thanks for helping me @barney2074 and @glenn-jocher

[Raymon360]

Yolov5/Ultralytics documentation is pretty good- I'd start by reading it

When I try it, the system can only work on 1 folder. How to make the system work if you have a lot of subfolders?
For example: "folders/subfolders/files"

[glenn-jocher]

@Raymon360 google glob documentation if you're not familiar with it

[github-actions]

👋 Hello, this issue has been automatically marked as stale because it has not had recent activity. Please note it will be closed if no further activity occurs.

Access additional YOLOv5 🚀 resources:

• Wiki – https://github.com/ultralytics/yolov5/wiki
• Tutorials – https://docs.ultralytics.com/yolov5
• Docs – https://docs.ultralytics.com

Access additional Ultralytics ⚡ resources:

• Ultralytics HUB – https://ultralytics.com/hub
• Vision API – https://ultralytics.com/yolov5
• About Us – https://ultralytics.com/about
• Join Our Team – https://ultralytics.com/work
• Contact Us – https://ultralytics.com/contact

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐!