实例分割Yolact边缘端部署 (四) 利用预训练模型快速标注
快速标注流程
- 预训练模型 如果数据集类别在coco类别内,那可以直接下载yolact的预训练模型
yolact_resnet50_54_800000.pth。如果是新的类别,那可以进行少量标注,然后训练一个初始版model。- 利用预训练模型对数据集图像进行推理,并将推理结果保存成json格式。
1. yolact推理流程
1.1 evalimage()
- 首先定位yolact推理单张图像的流程,利用
eval.py加载模型对图像进行推理,从函数def evalimage可以看出,preds = net(batch)就获得了推理结果,prep_display()对结果进行了显示。
def evalimage(net:Yolact
本文介绍了如何利用Yolact预训练模型对数据集进行快速标注,包括理解Yolact的evalimage()和prep_display()推理流程,增加label导出函数,以及推理并生成json标签的详细步骤。
订阅专栏 解锁全文
524

被折叠的 条评论
为什么被折叠?



