Fix relay.testing.darknet convert_image (apache#7667)

trevor-m · May 6, 2021 · 0f2bd24 · 0f2bd24
1 parent 8b5e094
commit 0f2bd24
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/testing/darknet.py b/python/tvm/relay/testing/darknet.py
@@ -31,7 +31,7 @@
 def convert_image(image):
     """Convert the image with numpy."""
     imagex = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-    imagex = np.array(image)
+    imagex = np.array(imagex)
     imagex = imagex.transpose((2, 0, 1))
     imagex = np.divide(imagex, 255.0)
     imagex = np.flip(imagex, 0)