作者您好，请问一下可视化的测试脚本中使用的模型权重文件及模式是什么配置呢？我用yaml里默认的配置会将brown属性错误的预测为wooden属性，与您可视化的图片结果差距较大

MILVLG / bottom-up-attention.pytorch

A PyTorch reimplementation of bottom-up-attention models

Apache License 2.0

294 stars 76 forks source link

作者您好，请问一下可视化的测试脚本中使用的模型权重文件及模式是什么配置呢？我用yaml里默认的配置会将brown属性错误的预测为wooden属性，与您可视化的图片结果差距较大 #95

Open LittleGrey-John opened 2 years ago

Yevet commented 2 years ago

可以问下您是怎么运行可视化脚本的吗？我发现运行到model = DefaultTrainer.build_model(cfg)的时候就会报错TypeError: from_config() missing 1 required positional argument: 'input_shape'或者TypeError: init () got an unexpected keyword argument 'first_stride'，不知道您遇到过这个问题吗？

LittleGrey-John commented 2 years ago

我是直接运行的并没有遇到您说的这个问题那你可以把input_shape加上 first_stirde去掉试试