水果数据集介绍
该数据集包括 6 种不同水果（苹果、葡萄、菠萝、橙子、香蕉和西瓜）的 8479 张图像。
水果以 YOLOv8 格式标注。
对每个图像应用以下预处理：
- 像素数据的自动定位（使用 EXIF 方向剥离）
- 调整为 640x640（拉伸）
训练代码介绍
SuperGradients¶
SuperGradients 是一个基于 PyTorch 的训练库。
它为最常见的计算机视觉用例提供统一的接口：
- 分类
- 检测
- 分割
- 姿势估计
我们的模型库中有近 40 个预训练模型。您可以通过此链接查看可用的预训练模型。
本笔记本将专注于将 SuperGradients 与 YOLO-NAS 结合使用。如果您有兴趣了解 SG 如何用于图像分类，您可以查看这个模板化的笔记本，它可以让您轻松上手。
第一步：库安装
%%capture
pip install super-gradients==3.1.0第一步：库导入
from PIL import Image
import os
import numpy as