PaddlePaddle / PaddleSlim

PaddleSlim is an open-source library for deep model compression and architecture search.
https://paddleslim.readthedocs.io/zh_CN/latest/
Apache License 2.0
1.56k stars 345 forks source link

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

Open sgAlbanC opened 7 months ago

sgAlbanC commented 7 months ago

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

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

image

zzjjay commented 7 months ago

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

sgAlbanC commented 7 months ago

嗯嗯好的,我知道input name的问题。但我在配置文件中如何修改呢。

这种配置文件中的关键字,以及配置文件的选项一般在哪里找到呀? 我平时在使用这些配置文件的时候,有些配置文件我并不了解其意义,有些属性我也不确定有哪些值可以选择。

是在Global中添加 input_name属性吗?

390200763 @.***

 

------------------ 原始邮件 ------------------ 发件人: "PaddlePaddle/PaddleSlim" @.>; 发送时间: 2024年4月8日(星期一) 下午5:35 @.>; @.**@.>; 主题: Re: [PaddlePaddle/PaddleSlim] 关于自动压缩yolov8-s,run.py的时候出错。 (Issue #1865)

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

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

zzjjay commented 7 months ago

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

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