使用 YOLO-NAS 目标检测训练从航拍图像中进行车辆检测

        文末包含源码+相关测试数据集的下载链接

        航拍影像为远距离对地目标观测提供了丰富的数据信息,尤其是随着无人机的发展,越来越多的航拍影像数据可被获取,为该场景下的目标检测与识别提供了坚实的数据基础。车辆识别作为航拍影像场景下的目标检测研究的重要方向之一,是智能交通系统中的重要一环,为路况实时获取、事故监测、违章停车监控等应用场景提供技术支撑。本文将介绍通过最新的YOLO-NAS高性能模型对航拍的图片进行汽车检测,本文同时提供相关的测试代码和数据集链接

安装

注意:安装完成后(可能需要几分钟),您需要在安装完成后重新启动运行时。

%%capture

!pip install super-gradients==3.1.0

必要库导入 

import os

import requests
import torch
from PIL import Image

from super_gradients.training import Trainer, dataloaders, models
from super_gradients.training.dataloaders.dataloaders import (
    coco_detection_yolo_format_train, coco_detection_yolo_format_val
)
from supe

猜你喜欢

转载自blog.csdn.net/tianqiquan/article/details/131064879#comments_27175906
今日推荐