Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

关于自动压缩yolov8-s,run.py的时候出错。 #1865

Open
sgAlbanC opened this issue Apr 1, 2024 · 3 comments
Open

关于自动压缩yolov8-s,run.py的时候出错。 #1865

sgAlbanC opened this issue Apr 1, 2024 · 3 comments
Assignees

Comments

@sgAlbanC
Copy link

sgAlbanC commented Apr 1, 2024

版本:paddleslim2.5 paddledet2.6 paddleYOLO2.5训练出来的 yolov8s模型,根据示例进行自动压缩出错。

我模型导出来,模型输入 为image,这里报错,说data没有image这个属性;该如何修改??

image

@zzjjay
Copy link
Collaborator

zzjjay commented Apr 8, 2024

可以通过netron.app 可视化pdmodel文件,查看模型 input name的名字,再在config中做出相应修改~

@sgAlbanC
Copy link
Author

sgAlbanC commented Apr 8, 2024 via email

@zzjjay
Copy link
Collaborator

zzjjay commented Apr 8, 2024

这个config是指自动压缩的配置文件,一般是在config目录下的yaml文件,其中Global里会有input name的配置;

如果没有,则是代码会自动根据模型input name进行配置,这时候可以在run.py出错的位置,打印input name,检查是否是dataloader读取的dict数据,没有相应的key值,然后再去相应修改dataloader读取的数据内容;

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants