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)
|