yolov5
目前内置yolov5模块提供了两个插件
# 教主yolo:
感谢作者教主提供的yolov5插件,此插件提供了一键部署yolov5训练环境,操作极其简单,插件执行速度在cpu下可以达到100-200毫秒
# tflite通用yolo模型
aiwork平台提供了通用yolov5插件,可以自行下载开源训练库或者把你已经训练好的yolov5模型转成.tflite格式即可使用
调用案例:
复制var yolo=new yoloV5();
//可以自己配置是否gpu加速和int8量化
yolo.loadTFMode("/插件/fp16_320.tflite","/插件/YoloV5.txt",320,false,false,false);
//或者简写
//yolo.loadTFMode("/插件/fp16_320.tflite","/插件/YoloV5.txt",640);
//本地图片识别
var img=new image().readToBitmap(project.getPluginsPath()+"/2.png");
//截图识别
// var img=screen.screenShotFull();
var result=yolo.detectTFlite(img);
printl(result);
如何训练yolo
可以直接使用tf训练后直接打包tflite
参考地址:https://github.com/yyccR/yolov5_in_tf2_keras (opens new window)
或者使用pytorch训练完以后再用模型量化转换工具转成tflite格式
参开地址:https://gitee.com/song-laogou/yolov5-mask-42 (opens new window)
上次更新: 2024/11/03, 18:44:54