Comments (16)
@yl305237731 I got up in the morning and looked at the training results. There is no problem with the EfficientNet model. At present, the mAP of Epoch35/49 has reached 0.982. Thank you for your contribution.
from flexible-yolov5.
@Zpadger i have try efficientnet B2 as backbone, it's no problem.
from flexible-yolov5.
@yl305237731 I just located the problem, and it seems I know the reason, I forgot to change the classes in the model_efficient.yaml file,so the "nc" is still "nc : 1".I have corrected the value now, and I am going to continue to run the experiment and wait for tomorrow to verify the result.
from flexible-yolov5.
@yl305237731 I also encountered the same problem, using EfficientNet:
python ./scripts/train.py --img 640 --batch 4 --epoch 10 --data ./data/color_class.yaml --cfg ./configs/model_efficientnet.yaml
~
@Zpadger 😳,OK ,i will try this backbone.
from flexible-yolov5.
Don't know if anything I need to be notified of before training with code in this repo....
from flexible-yolov5.
@ShirleyHe2020 can you tell me which backbone ? I will verify it.
from flexible-yolov5.
@yl305237731 I also encountered the same problem, using EfficientNet:
python ./scripts/train.py --img 640 --batch 4 --epoch 10 --data ./data/color_class.yaml --cfg ./configs/model_efficientnet.yaml
~
from flexible-yolov5.
Don't know if anything I need to be notified of before training with code in this repo....
@ShirleyHe2020 hi, may be your problem is same as @Zpadger , so check nc in model_XXXX.yaml.
from flexible-yolov5.
@ShirleyHe2020 can you tell me which backbone ? I will verify it
from my side, all the backbones ( mobilenet, efficientnet, reset ) suffer the same issue.
from flexible-yolov5.
@ShirleyHe2020 i think you forget to change the nc to your dataset in model_xxx.yaml
from flexible-yolov5.
Don't know if anything I need to be notified of before training with code in this repo....
@ShirleyHe2020 hi, may be your problem is same as @Zpadger , so check nc in model_XXXX.yaml.
thanks @Zpadger , same issue , 'nc' changed then mAP reached expected level
from flexible-yolov5.
Hi @ShirleyHe2020 @Zpadger Where to load the efficientnet weights when using the B6 as backbone? It does not remind me to download weights or load it.
from flexible-yolov5.
@Lg955 hi, i have try b6, there is no problem
add pretrained: True in model_efficientnet.yaml under backbone
from flexible-yolov5.
@yl305237731 My showing of result is different from yours: (the error in the bottom of img is out of cuda memory, ignore it)
And the .yaml is this:
backbone: type: efficientnet version: b2 # b1 to b8 and l2 head: nc: 5 stride: [8.0, 16.0, 32.0] 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
from flexible-yolov5.
@Lg955 for above information, I can't see the problem. and i test b2, there is no problem
from flexible-yolov5.
Hi, @yl305237731 when detect imgs, the pred result is empty after NMS(pred result is OK before NMS), have you encountered the same problem?
`pred before NMS tensor([[[7.02863e+00, 4.92409e+00, 2.66565e+01, ..., 3.92634e-02, 1.15775e-01, 2.56261e-03],
[1.08675e+01, 4.00093e+00, 3.37639e+01, ..., 4.27754e-02, 8.12325e-02, 4.94392e-03],
[1.85410e+01, 3.28052e+00, 3.73997e+01, ..., 5.11962e-02, 1.00191e-01, 5.55359e-03],
...,
[1.20430e+03, 7.04980e+02, 2.17258e+02, ..., 3.21729e-02, 2.10942e-01, 8.92933e-02],
[1.22968e+03, 7.08009e+02, 2.33180e+02, ..., 4.42553e-02, 2.42864e-01, 1.20849e-01],
[1.25142e+03, 7.13127e+02, 1.60746e+02, ..., 3.88758e-02, 1.66492e-01, 1.16945e-01]]], device='cuda:0')
/------------------------/
pred after NMS [tensor([], device='cuda:0', size=(0, 6))]`
from flexible-yolov5.
Related Issues (20)
- 数据增强 HOT 1
- 更改backbone HOT 1
- 有关head的问题 HOT 3
- 请问yaml文件中有关head的网络结构 HOT 4
- 请问模型的flops应该在哪输出 HOT 3
- 作者大大是否有跟进YOLOv5 V7.0版本实例分割功能的打算? HOT 2
- About the Resume HOT 1
- 使用mobilenetv3作为主干网络,无法导出onnx,并且没有错误提示。 HOT 5
- Add True negative data to training dataset HOT 2
- 使用替换backbone为swimtransformer的yolov5时,detect的时候,只能框住物体,不能显示置信度和种类 HOT 2
- ONNX: export failure: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument other in method wrapper__equal) HOT 5
- error during compile HOT 11
- 求教 HOT 1
- Drop layer HOT 1
- Error help please HOT 3
- repvgg做为backbone时,保存的pt模型是多分支结构还是单路结构? HOT 5
- 使用repvgg训练,然后参数deploy为true,即改成单路推理结构时,出现训练无法收敛的情况?是不是由于每一次训练都直接转成了单路结构的原因? HOT 30
- delete
- Quantization error HOT 1
- SE and CBAM module implementation error
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 flexible-yolov5.