24bf232753b4b34b6d4164d5a647dd17d16348b2,example/convert_resnet/convert.py,,main,#,23

Before Change

    graph_exec_data.save(OUTPUT_DIR)

    with open(path.join(OUTPUT_DIR, "image_nhwc.json".format()), "w") as f:
        image_nhwc = np.transpose(prepared_image, (1, 2, 0))
        json.dump(image_nhwc.flatten().tolist(), f)

    with open(path.join(OUTPUT_DIR, "fc6.json".format()), "w") as f:
        json.dump(nn_output.data.tolist(), f)

After Change

    parser.add_argument("--encoding")
    args = parser.parse_args()

    sample_image = np.zeros((224, 224, 3), dtype=np.uint8)//PIL.Image.open("")
    if args.model == "vgg16":
        link = chainer.links.model.vision.vgg.VGG16Layers()
        prepared_image = chainer.links.model.vision.vgg.prepare(sample_image)  // BGR, CHW

Instances