search

fuqianya / AoANet-Paddle

3 stars 0 forks source link

# load pre-extract image features #2

Open cylvzj opened 2 years ago

cylvzj commented 2 years ago

你好，请教一下。 在predict.py 中预测是这样的： feats = np.load('data/573223.npz') 但实际用户输入的是一张图片， 请问一下怎样修改predict.py，以图片路径为参数传入？ 或者说怎么将image 转换成 npz格式？

谢谢，在线等待回复

fuqianya commented 2 years ago

是的，573223.npz在data文件下有: https://github.com/fuqianya/AoANet-Paddle/blob/main/data/573223.npz

cylvzj commented 2 years ago

573223.npz这个文件找到了， 但用户输入的是一张随手拍的图片，

cylvzj commented 2 years ago

就是怎么使用 predict.py， 参数是一张图片， 不是固定的data/573223.npz

应该有两种途径： 1、先把图片转换成 npz， 在调用 2、把predict.py 修改一下，把参数改成图像输入，

fuqianya commented 2 years ago

对于任何一张图像的输入，可以提取图像的bottom-up特征（即npz文件），提取方法见: https://github.com/peteanderson80/bottom-up-attention

cylvzj commented 2 years ago

我之前也找到过这里，测试了 extract_feat.py 和 extract_feat_v2.py ， 但没有GPU， 运行失败了， 怎样修改成用CPU了