ModelScope训练出来的模型预测,修改damoyolo.py里的model.name和Zero

ModelScope训练出来的模型预测,修改damoyolo.py里的model.name和ZeroHead.classes这两个参数后还需要修改其他的吗?目前我用一个小样本dog1训练后,识别出的标签为什么是person,是不是有些地方需要修改创业?

展开
收起
真的很搞笑 2023-04-10 23:29:03 180 分享 版权
1 条回答
写回答
取消 提交回答
  • 还需要修改damoyolo.py中的model字段下的coco_label_map.pkl文件,这里面存储了类别到index的映射,您需要根据您的数据集映射,手动修改该文件。,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-04-11 19:10:08
    赞同 展开评论

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

收录在圈子:
+ 订阅

热门讨论

热门文章

还有其他疑问?
咨询AI助理