Update app.py
Browse files
app.py
CHANGED
@@ -106,7 +106,7 @@ def get_model(ctx, model_path):
|
|
106 |
mx.test_utils.download('https://s3.amazonaws.com/onnx-model-zoo/duc/city1.png')
|
107 |
|
108 |
# read image as rgb
|
109 |
-
im = cv.imread(
|
110 |
# set output shape (same as input shape)
|
111 |
result_shape = [im.shape[0],im.shape[1]]
|
112 |
# set rgb mean of input image (used in mean subtraction)
|
|
|
106 |
mx.test_utils.download('https://s3.amazonaws.com/onnx-model-zoo/duc/city1.png')
|
107 |
|
108 |
# read image as rgb
|
109 |
+
im = cv.imread('city1.png')[:, :, ::-1]
|
110 |
# set output shape (same as input shape)
|
111 |
result_shape = [im.shape[0],im.shape[1]]
|
112 |
# set rgb mean of input image (used in mean subtraction)
|