- 可以训练yolov5(v6.0)、yolox、小型网络，添加注意力机制
xialuxi / yolov5_all Goto Github PK
View Code? Open in Web Editor NEW可以训练yolov5(v6.0)、yolox、小型网络，添加注意力机制
License: GNU General Public License v3.0
yolov5_all's Introduction
yolov5_all's People
Contributors
Stargazers
Watchers
Forkers
xinsuinizhuan pudongdong shizhanhao zoey-smile egrass wstchhwp jimmyw-11 li33609 lifeng0718 zqyby geng5610248 bestsongc eamon-cai theteaterrayolov5_all's Issues
CBAM 没有导入头文件吗？
Downloading https://ultralytics.com/assets/Arial.ttf to /root/.config/Ultralytics/Arial.ttf...
Traceback (most recent call last):
File "/content/drive/MyDrive/yolov5_all-main/train.py", line 37, in
from models.yolo import Model
File "/content/drive/MyDrive/yolov5_all-main/models/yolo.py", line 288
MobileNetV3_Block, MobileNetV2_Block，CBAM]:
^
SyntaxError: invalid character in identifier
是不是应该在yolo.py中导入
from models.common import Conv, Bottleneck, SPP, DWConv, Focus, BottleneckCSP, Concat, CBAM
YOLOX无法训练
前辈您好 训练YOLOX的时候报错 请问您遇到过吗？
我是北矿的研究生 您方便留联系方式交流吗？感谢。
发现一个小问题
File "/content/drive/MyDrive/yolov5_all-main/train.py", line 278
if if is_parallel(model):
^
SyntaxError: invalid syntax
多GPU训练报错
File "yolov5_all-main/models/yolox.py", line 246, in get_output_and_grid
reg_box[..., :2] = (reg_box[..., :2] +grid)*stride # （第k层）预测物体的中心坐标
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0!
CBAM 对 mAP 的提升效果
我这边希望在 YOLOv3 的模型中增加 CBAM模块，参考您的yaml文件，设置如下
不知是否可行
另外，我很好奇，您插入的CBAM是否有跑过实验，有大约多大的提升
yoloxs.yaml
您好，为啥您的yoloxs.yaml相比官网的YOLOX主干在focus后面少了一个卷积，这还能算是YOLOX的网络结构吗
giou有一些小错误
引入CBAM对于map影响大吗
想知道对YOLOv5引入CBAM后对map的影响大吗？
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.