强化学习 yolov5
时间: 2023-09-13 10:04:19 浏览: 187
yolov5s模型,yolov5学习
引用:目前的 sahi 只支持 yolov5 和 mmdetection,但是,我们可以很方便的添加新框架的支持,你只需要在 sahi/sahi/model.py 文件中创建一个新的类,这个类继承自 DetectionModel,然后依次实现 load_model、perform_inference、_create_object_prediction_list_from_original_predictions、_create_original_predictions_from_object_prediction_list这几个方法即可,具体的可以参考 Yolov5DetectionModel 类的实现 关联阅读。
根据引用,强化学习 yolov5 在 sahi 项目中是被支持的。sahi 项目目前支持 yolov5 和 mmdetection 两种框架。如果你想要添加其他框架的支持,你只需要在 sahi/sahi/model.py 文件中创建一个新的类,继承自 DetectionModel,并实现相应的方法即可。可以参考 Yolov5DetectionModel 类的实现来进行操作。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [SAHI强化YOLOv5在小目标上的表现](https://blog.csdn.net/djstavaV/article/details/120559239)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]
阅读全文