Fix relay.testing.darknet convert_image (#7667)

apache · Mar 16, 2021 · e697f03 · e697f03
1 parent b8b63cd
commit e697f03
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)