改进yolov5(多检测头+注意力机制+repvgg结构)pytorch源码+项目说明.zip
集成yolov5(v6.0), 注意力机制， 和repvgg结构
添加了多头检测代码，使用train_multiple_detection_head.py文件进行训练
添加了检测+关键点代码，使用train_key_point.py文件进行训练
2022-12-07 12:27:46
46.01MB
所谓“VGG式”指的是：
1. 没有任何分支结构。即通常所说的plain或feed-forward架构。
2. 仅使用3x3卷积。
3. 仅使用ReLU作为激活函数。
详见：
https://blog.csdn.net/hhhhhhhhhhwwwwwwwwww/article/details/126516646
2022-10-12 17:07:00
986.61MB
RepVGG：使VGG样式的ConvNets再次变得出色（CVPR-2021）（PyTorch）
这是一个超级简单的ConvNet架构，通过3x3 conv和ReLU的堆栈在ImageNet上实现了80％的top-1精度！ 此存储库包含预训练的模型，用于构建模型，训练的代码以及从训练时模型到推理时的转换。
MegEngine版本： : 。
@upczww 使用C ++ API进行TensorRT实现。 做得好！
@zjykzj 另一个不错的PyTorch实现。
包含在一个著名的模型动物园中（超过7k星） 。
更新（2021年1月13日）：您可以随时以不同的方式获得等效的内核和偏差（repvgg.py中的get_equivalent_kernel_bias）。 这可能有助于基于训练的修剪或量化。
更新（2021年1月31日）：此训练脚本（超简单的PyTorch官方示例样式
2021-05-09 22:21:38
301KB
1