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
Oliver6565 opened this issue Apr 1, 2024 · 3 comments
Open

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

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

Comments

@Oliver6565
Copy link

版本: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中做出相应修改~

@Oliver6565
Copy link
Author

Oliver6565 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