Update visualization_gary.py
Browse files- visualization_gary.py +1 -1
visualization_gary.py
CHANGED
|
@@ -319,7 +319,7 @@ def direct_inference(input):
|
|
| 319 |
Input=tr(Input)
|
| 320 |
image_np = (Input * 255).clamp(0, 255).byte()
|
| 321 |
image_np = image_np.permute(1, 2, 0).numpy()
|
| 322 |
-
image_np = cv2.cvtColor(image_np, cv2.COLOR_RGB2BGR)
|
| 323 |
|
| 324 |
|
| 325 |
ORI= overlapping_features_on_input(model,output, feature_maps, image_np,output)#input image_np
|
|
|
|
| 319 |
Input=tr(Input)
|
| 320 |
image_np = (Input * 255).clamp(0, 255).byte()
|
| 321 |
image_np = image_np.permute(1, 2, 0).numpy()
|
| 322 |
+
# image_np = cv2.cvtColor(image_np, cv2.COLOR_RGB2BGR)
|
| 323 |
|
| 324 |
|
| 325 |
ORI= overlapping_features_on_input(model,output, feature_maps, image_np,output)#input image_np
|