- 将yolov5的权重文件或者自定义权重文件(.pt)加入到yolov5的根目录
python export.py --weights yolov5s.pt --include tfjs
- 将
yolov5s_web_model
目录加入到static
中
📌 修改yolov5_rt_tfjs.js
文件,自定义权重文件和类别名称
// 权重文件
const weights = '../yolov5s_web_model/model.json';
// 类别名称(中文版)
const cls_names = ['人', '自行车', '汽车', '摩托车', '飞机', '公交车', '火车', '卡车', '船', '红绿灯', '消防栓', '停止标志',
'停车收费表', '长凳', '鸟', '猫', '狗', '马', '羊', '牛', '象', '熊', '斑马', '长颈鹿', '背包', '雨伞', '手提包', '领带',
'手提箱', '飞盘', '滑雪板', '单板滑雪', '运动球', '风筝', '棒球棒', '棒球手套', '滑板', '冲浪板', '网球拍', '瓶子', '红酒杯',
'杯子', '叉子', '刀', '勺', '碗', '香蕉', '苹果', '三明治', '橙子', '西兰花', '胡萝卜', '热狗', '比萨', '甜甜圈', '蛋糕',
'椅子', '长椅', '盆栽', '床', '餐桌', '马桶', '电视', '笔记本电脑', '鼠标', '遥控器', '键盘', '手机', '微波炉', '烤箱',
'烤面包机', '洗碗槽', '冰箱', '书', '时钟', '花瓶', '剪刀', '泰迪熊', '吹风机', '牙刷'
]
📌 修改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