| #!/usr/bin/python3 |
| """ Generate inputs for NNAPI benchamrks using a image |
| |
| Usage: |
| ./gen_mobilenet_input.py image_file.jpg |
| """ |
| |
| |
| from PIL import Image |
| import numpy as np |
| import sys |
| import os |
| |
| def gen_input_files(filename, prefix, size): |
| basename = os.path.basename(filename) + "_" + prefix |
| img_f64 = np.array(Image.open(filename).resize(size)).astype(np.float) / 128 - 1 |
| with open(basename + "_f32.bin", "wb") as f: |
| f.write(img_f64.astype('float32').tobytes()) |
| print("Saving: " + basename + "_f32.bin") |
| with open(basename + "_u8.bin", "wb") as f: |
| f.write(((img_f64 + 1) * 255).astype('uint8').tobytes()) |
| print("Saving: " + basename + "_u8.bin") |
| |
| if __name__ == '__main__': |
| if len(sys.argv) < 1: |
| print("Usage:\n ./gen_mobilenet_input.py image_file.jpg\n") |
| sys.exit(1) |
| |
| gen_input_files(sys.argv[1], "mobilenet", (224, 224)) |
| gen_input_files(sys.argv[1], "inceptionv3", (299, 299)) |
| gen_input_files(sys.argv[1], "hdrnet", (256, 256)) |