游戏人生
首页
(current)
GameDevTools
登陆
|
注册
个人中心
注销
ImGui-WebAssembly
Introduction
Introduction
前言
前言
1. SuperVision介绍
1. SuperVision介绍
1.1 使用Box和Label标记图片物体
1.2 导出检测结果
1.3 过滤检测结果
1.4 检测小物体
1.5 踪视频中的对象
1.6 处理数据集
代码资源下载
点我下载
Github
点赞、收藏、关注
目录
<< 1.1 使用Box和Label标记图片物体
1.3 过滤检测结果 >>
## 导出检测结果 ```text 「supervision book」是一本开源电子书,PDF/随书代码/资源下载: https://github.com/ThisisGame/supervision_book ``` 可以导出物体检测结果到csv、json、自定义结构。 官方文档 `https://supervision.roboflow.com/latest/how_to/save_detections/` ### 准备视频 先下载测试视频。 ```python #file:files\supervision\download_people_walking_mp4.py from supervision.assets import download_assets, VideoAssets download_assets(VideoAssets.PEOPLE_WALKING) ``` ### 导出到csv ```python #file:files\supervision\save_detections_to_csv.py import supervision as sv from ultralytics import YOLO model = YOLO("yolov8n.pt") frames_generator = sv.get_video_frames_generator("people-walking.mp4") with sv.CSVSink("./save_detactions_to_csv.csv") as sink: for frame in frames_generator: results = model(frame)[0] detections = sv.Detections.from_ultralytics(results) sink.append(detections, {}) ``` 执行后就会将检测结果输出到`./save_detactions_to_csv.csv`,如下图: <a id="antiCollectorAdTxt" href="https://github.com/ThisisGame/supervision_book">「supervision book」是一本开源电子书,PDF/随书代码/资源下载: https://github.com/ThisisGame/supervision_book</a> ### 导出自定义列 上面直接导出的csv,缺少了帧序号,下面就用自定义列来添加帧序号。 ```python #file:files\supervision\save_detections_to_csv_custom_column.py import supervision as sv from ultralytics import YOLO model = YOLO("yolov8n.pt") frames_generator = sv.get_video_frames_generator("people-walking.mp4") with sv.CSVSink("./save_detections_to_csv_custom_column.csv") as sink: for frame_index, frame in enumerate(frames_generator): results = model(frame)[0] detections = sv.Detections.from_ultralytics(results) sink.append(detections, {"frame_index": frame_index})#用自定义列来添加帧序号 ```  ### 导出到Json ```python #file:files\supervision\save_detections_to_json.py import supervision as sv from ultralytics import YOLO model = YOLO("yolov8n.pt") frames_generator = sv.get_video_frames_generator("people-walking.mp4") with sv.JSONSink("./save_detections_to_json.json") as sink: for frame_index, frame in enumerate(frames_generator): results = model(frame)[0] detections = sv.Detections.from_ultralytics(results) sink.append(detections, {"frame_index": frame_index}) ``` 
<< 1.1 使用Box和Label标记图片物体
1.3 过滤检测结果 >>
12
代码资源下载
点我下载
Github
点赞、收藏、关注
目录
Introduction
Introduction
前言
前言
1. SuperVision介绍
1. SuperVision介绍
1.1 使用Box和Label标记图片物体
1.2 导出检测结果
1.3 过滤检测结果
1.4 检测小物体
1.5 踪视频中的对象
1.6 处理数据集