search

openppl-public / ppq

PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.
Apache License 2.0
1.44k stars 223 forks source link

ProgramEntrance_2.py读取imagenet图片reshape报错 #565

Open huanFigure opened 3 weeks ago

huanFigure commented 3 weeks ago

ProgramEntrance_2.py读取imagenet图片报错，reshape失败 arr = np.fromfile(path, dtype=np.dtype('float32')).reshape([1, 3, 224, 224]) 因为reshape不可改变数据的元素数量，所以如果imagenet的图片大小与reshape的值不匹配会报错，所以示例代码应该先处理图片大小格式为3，224，224，再转为np的数组