Comments (5)
from yolov5prune.
from yolov5prune.
yolov5s_bifpn.yaml结构如下:
parameters
nc: 2 # number of classes
depth_multiple: 0.33 # model depth multiple
width_multiple: 0.50 # layer channel multiple
anchors
anchors:
- [10,13, 16,30, 33,23] # P3/8
- [30,61, 62,45, 59,119] # P4/16
- [116,90, 156,198, 373,326] # P5/32
YOLOv5 backbone
backbone:
[from, number, module, args]
[[-1, 1, Focus, [64, 3]], # 0-P1/2
[-1, 1, Conv, [128, 3, 2]], # 1-P2/4
[-1, 3, C3, [128]],
[-1, 1, Conv, [256, 3, 2]], # 3-P3/8
[-1, 9, C3, [256]],
[-1, 1, Conv, [512, 3, 2]], # 5-P4/16
[-1, 9, C3, [512]],
[-1, 1, Conv, [1024, 3, 2]], # 7-P5/32
[-1, 1, SPP, [1024, [5, 9, 13]]],
[-1, 3, C3, [1024, False]], # 9
]
YOLOv5 head
head:
[[-1, 1, Conv, [512, 1, 1]],
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
[[-1,6], 1, Concat_bifpn, [256,256]], # cat backbone P4
[-1, 3, C3, [512, False]], # 13
[-1, 1, Conv, [256, 1, 1]],
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
[[-1, 4], 1, Concat_bifpn, [128,128]], # cat backbone P3
[-1, 3, C3, [256, False]], # 17 (P3/8-small)
[-1, 1, Conv, [512, 3, 2]], # 320, 640 #
[[-1, 6, 13], 1, Concat_bifpn, [256,256]], # cat head P4
[-1, 3, C3, [512, False]], # 20 (P4/16-medium)
[-1, 1, Conv, [1024, 3, 2]], # 640, 1280 #
[[-1, 9], 1, Concat_bifpn, [512, 512]], # cat head P5 cat 20,20 #22
[-1, 3, C3, [1024, False]], # 25 (P5/32-large) # 1280, 1280 #23
[[17, 20, 23], 1, Detect, [nc, anchors]] # Detect(P3, P4, P5)
]
from yolov5prune.
Hello @Hazel-Xiang , how did you finish the pruning? What is the code modification idea?
from yolov5prune.
你好,请问你有解决方法了吗?我现在也困在这里
from yolov5prune.
Related Issues (20)
- 在进行稀疏化的时候遇到报错 HOT 2
- TypeError: on_fit_epoch_end() missing 1 required positional argument: 'fi'
- 6.0版本稀疏训练train_sparity.py 报错TypeError: on_fit_epoch_end() takes 5 positional arguments but 6 were given HOT 2
- 我想请教下稀疏训练的稀疏效果一直不好是怎么回事,我甚至把sr开到了10 HOT 2
- prune.py脚本运行报错 too many indices for tensor demension 3 HOT 1
- 计算量 HOT 1
- RuntimeError: mat1 and mat2 shapes cannot be multiplied HOT 1
- 训练自己的算法
- 使用train_sparsity.py出现的错误- HOT 1
- 已经充分稀疏的情况下,剪枝率只能设置到0.1
- 关于train.py训练后tensorboard中没有bn_weights/hist直方图 HOT 1
- v7.0版本能用吗? HOT 4
- 运行预测脚本的时候报错了 显示yolo脚本没有DetectionModel这个属性
- 更换主干网络再经行稀疏化训练后,值变成了nan
- tensorboard中的histogram图表中的数据如何导出呢
- IndexError: too many indices for tensor of dimension 3
- 稀疏因子最高只能设为0.1,再高一点剪枝最大比例反而会降低是怎么回事
- 剪枝出现Scanning 'E:\task\experiment\dataset\BITVehicle\BITVehicle\val.cache' images and labels... 0 found, 985 missing, 0 empty, 0 corrupted HOT 1
- 博主麻烦看看这个问题 进行prune.py 出现KeyError: 'model.0.bn'问题
- Can't get attribute 'ModelPruned' on <module 'models.yolo'
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from yolov5prune.