diff --git a/tutorials/nnvm/deploy_ssd.py b/tutorials/nnvm/deploy_ssd.py index 45fea1985b166..cf950d5b85ea7 100644 --- a/tutorials/nnvm/deploy_ssd.py +++ b/tutorials/nnvm/deploy_ssd.py @@ -39,21 +39,14 @@ # x86 conv2d schedule doesn't support dilation. supported_model = [ - 'ssd_512_resnet18_v1_voc', - 'ssd_512_resnet18_v1_coco', - 'ssd_512_resnet18_v1_custom', 'ssd_512_resnet50_v1_voc', 'ssd_512_resnet50_v1_coco', - 'ssd_512_resnet50_v1_custom', 'ssd_512_resnet101_v2_voc', - 'ssd_512_resnet152_v2_voc', - 'ssd_512_mobilenet1_0_voc', - 'ssd_512_mobilenet1_0_coco', - 'ssd_512_mobilenet1_0_custom', + 'ssd_512_mobilenet1.0_voc', + 'ssd_512_mobilenet1.0_coco', ] model_name = "ssd_512_resnet50_v1_voc" -test_image = "dog.jpg" dshape = (1, 3, 512, 512) dtype = "float32" target = "llvm" @@ -86,12 +79,12 @@ # execute m.run() # get outputs -class_IDs, scores, bounding_boxs = tvm.get_output(0), tvm.get_output(1), tvm.get_output(2) +class_IDs, scores, bounding_boxs = m.get_output(0), m.get_output(1), m.get_output(2) ###################################################################### # Display result -ax = utils.viz.plot_bbox(img, bounding_boxs[0], scores[0], - class_IDs[0], class_names=block.classes) +ax = utils.viz.plot_bbox(img, bounding_boxs.asnumpy()[0], scores.asnumpy()[0], + class_IDs.asnumpy()[0], class_names=block.classes) plt.show()