热门
ModelScope训练出来的模型预测,修改damoyolo.py里的model.name和ZeroHead.classes这两个参数后还需要修改其他的吗?目前我用一个小样本dog1训练后,识别出的标签为什么是person,是不是有些地方需要修改创业?
还需要修改damoyolo.py中的model字段下的coco_label_map.pkl文件,这里面存储了类别到index的映射,您需要根据您的数据集映射,手动修改该文件。,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”
包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域