Haaribo commited on
Commit
977ee66
·
verified ·
1 Parent(s): bfe9b4e

Update visualization_gary.py

Browse files
Files changed (1) hide show
  1. 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