目标检测模型在pytorch当中的实现一个yolov7的库可以用于训练自己的数据集
yolov7训练自己数据集支持step cos学习率下降法支持adam sgd优化器选择支持学习率根据batch size自适应调整新增图片裁剪支持多GPU训练支持各个种类目标数量计算支持heatmap支持EMA.
文件列表
yolov7-pytorch-master.zip
(预估有个34文件)
yolov7-pytorch-master
voc_annotation.py
7KB
kmeans_for_anchors.py
6KB
LICENSE
34KB
predict.py
9KB
utils
utils.py
3KB
__init__.py
1B
utils_bbox.py
21KB
utils_map.py
36KB
暂无评论