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
|