Git Product home page Git Product logo

halotrouvaille / yolo-multi-backbones-attention Goto Github PK

View Code? Open in Web Editor NEW
493.0 493.0 118.0 1.15 MB

Model Compression—YOLOv3 with multi lightweight backbones(ShuffleNetV2 HuaWei GhostNet), attention, prune and quantization

Home Page: https://github.com/HaloTrouvaille/YOLO-Multi-Backbones-Attention

Python 98.22% Shell 1.78%
attention-mechanism cbam eca huawei-ghostnet object-detection-datasets se shufflenet yolo

yolo-multi-backbones-attention's People

Contributors

halotrouvaille avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

yolo-multi-backbones-attention's Issues

Error while training with a custom dataset

Hello I am trying to train with a custom dataset and I am facing the following issue:

image

All images of my custom dataset is 448x320. I already made the changes in image size parameter, but this error still showing.

Could someone help me with this issue?

train.py

Traceback (most recent call last):
File "train.py", line 467, in
opt.cfg = list(glob.iglob('./**/' + opt.cfg, recursive=True))[0] # find file
IndexError: list index out of range
Why?

“activation=linear”

“activation=linear” appears in cfg files, but there is no description in “models.py”. Can you tell me what to do?

关于cfg

你好,关于cfg文件我有个小问题:我看到原始yolov3的cfg文件中第一个卷积核的数量是64,而你的ghostnet的cfg文件中第一个卷积核数量是16,而且基本上每一个卷积核的数量都比原始的要少,这是为什么呢?

cfg文件的问题

在吗作者 有一些问题可否问下。cfg文件中的内容完整吗?

VoVNet

骨干网络为VoVNet的cfg文件以及对应的model.py网络解析内容大佬有吗??

量化问题

你好,量化前后为什么保存的模型大小都是180MB,量化并没有减小模型的大小?

where is cbam?

I only saw se,but no cbam,please tell me where is it,thank you!

RuntimeError: Sizes of tensors must match except in dimension 2. Got 40 and 48

Thank you for your great work.An error occurred while I was running the code.
python3 train.py --data data/oxfordhand.data --batch-size 1 --cfg cfg/yolov3-tiny-hand-se.cfg --epochs 100 -sr --s 0.001 --img-size 640 --weights ''
......
RuntimeError: Sizes of tensors must match except in dimension 2. Got 40 and 48
Can you give me some advice to solve this problem?

#数据集

你好作者,想问下有没有bdk100的数据集网盘呀,分享的链接取消了,感谢!!

量化问题

您好, 代码里量化时没有对BN层做处理, 参数仍然是fp32, 这个在部署时要怎么处理呢?

visdrone训练参数

您好,您给出的visdrone数据集的训练结果就是您代码中的默认参数的结果吗

Train erro

您好,我在用yolov3-tiny训练自己的数据集时出现了以下错误,能帮忙看下么
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 40 and 48 in dimension 2 at /pytorch/aten/src/THC/generic/THCTensorMath.cu:71

Parameters for Ghostnet + YoloV3

Hi,
You have mentioned the number of parameters for Ghostnet + Yolov3 to be 23.49M. I wanted to confirm if it is correct or it should be 2.349M? The number of parameters seem to be too larger than the other models.

Thanks.

问题请教

你换了ShuffleUnit之后,除了模型方面创建方面,其他地方有没有修改,我的微信17868380981,我有个疑问想和你讨论。

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo 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.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.