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

neo-ai · May 11, 2021 · a698edb · a698edb
1 parent 060105f
commit a698edb
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)