前言:本期我向大家介绍一个比较小众的模型训练软件——Lobe,这个软件可以让零基础小白快速上手分类视觉,例如什么垃圾分类之类。与大家所熟知的YOLO系列算法不同,检测算法通常包含分类,同时还有具体坐标输出。所以在分类原始数据集的处理上也存在差异。分类模型只需要把对应的小类放到一个文件夹下就行,并不需要打标处理。这里我先把链接放这里,之后再详细的写。Lobe (github.com)
1.软件
软件打开之后,就是如图所示
点击右上角“import”导入Dataset,这是一个文件夹
导入之后就可以点击右侧训练,等待训练好之后再点击“Use”即可导出模型文件,可以直接使用
然后就可以使用.onnx格式的模型文件了。