基于damoyolo,基于自己的数据集重新训练了一个检测模型,请问如何用pipeline加载?
以DAMO-YOLO-S为例, 1.您可以先调用一遍DAMO-YOLO-S的pipeline,
找到pipeline推理时的默认缓存路径,如果您是linux系统,该默认路径应该是/home/user_name/.cache/modelscope/hub/damo/cv_tinynas_object-detection_damoyolo在该路径下将damoyolo.py文件中的model.name字段,ZeroHead.num_classes字段替换成您的模型。再执行一遍步骤1中的推理代码,此时就是调用您自己训练的模型进行推理了。
赞0
踩0