wangrongsheng / yolov5_tfjs

✅YOLOv5 TFjs infer demo

导出模型

1. 将yolov5的权重文件或者自定义权重文件（.pt）加入到yolov5的根目录

python export.py --weights yolov5s.pt --include tfjs

2. 将yolov5s_web_model 目录加入到static 中

自定义模型

📌 修改yolov5_rt_tfjs.js文件，自定义权重文件和类别名称

// 权重文件
const weights = '../yolov5s_web_model/model.json';

// 类别名称（中文版）
const cls_names = ['人', '自行车', '汽车', '摩托车', '飞机', '公交车', '火车', '卡车', '船', '红绿灯', '消防栓', '停止标志',
    '停车收费表', '长凳', '鸟', '猫', '狗', '马', '羊', '牛', '象', '熊', '斑马', '长颈鹿', '背包', '雨伞', '手提包', '领带',
    '手提箱', '飞盘', '滑雪板', '单板滑雪', '运动球', '风筝', '棒球棒', '棒球手套', '滑板', '冲浪板', '网球拍', '瓶子', '红酒杯',
    '杯子', '叉子', '刀', '勺', '碗', '香蕉', '苹果', '三明治', '橙子', '西兰花', '胡萝卜', '热狗', '比萨', '甜甜圈', '蛋糕',
    '椅子', '长椅', '盆栽', '床', '餐桌', '马桶', '电视', '笔记本电脑', '鼠标', '遥控器', '键盘', '手机', '微波炉', '烤箱',
    '烤面包机', '洗碗槽', '冰箱', '书', '时钟', '花瓶', '剪刀', '泰迪熊', '吹风机', '牙刷'
]

自定义TensorFlow.js版本

📌 修改index.html文件，自定义TensorFlow.js的版本

<script src="https://cdn.jsdelivr.net/npm/@tensorflow/[email protected]/dist/tf.min.js"></script>

❗ 注意：TensorFlow.js的版本>=3.9.0，本项目目前使用的版本为3.16.0

yolov5_tfjs's Issues

从yolov5转到tfjs后,置信度超级低

从yolov5转到tfjs后,置信度超级低,原本0.97的置信度到浏览器变成0.5左右了 而且范围也不全