大佬，如何Python 推理？

heyongxin233 / YOLO-Pruning-RKNN

Easy Training Official YOLOv8、YOLOv7、YOLOv6、YOLOv5 and Prune all_model using Torch-Pruning!

48 stars 6 forks source link

Closed huangwgang closed 10 months ago

heyongxin233 commented 10 months ago

你可以使用该项目中的infer.py推理，具体可以参考文档Predict，如果想使用瑞芯微的开发板推理，可以参考rknn_yolov8

huangwgang commented 10 months ago

大佬，单类别评估模型如何操作？

heyongxin233 commented 10 months ago

你说说的是把所有类别当成一类吗，你可以在训练的时候加入参数single_cls=True，具体你可以看https://docs.ultralytics.com/usage/cfg/#train