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

Update visualization_gary.py

Browse files
Files changed (1) hide show
  1. visualization_gary.py +1 -0
visualization_gary.py CHANGED
@@ -208,6 +208,7 @@ def get_options_from_trainingset(output, model, TR, device,with_other_class):
208
 
209
  image = cv2.imread(f"options/{t}.jpg")
210
  concatenate_class = np.array(image)
 
211
 
212
  op.append((concatenate_class,features_id_neg))# intergrate negative features
213
  return op
 
208
 
209
  image = cv2.imread(f"options/{t}.jpg")
210
  concatenate_class = np.array(image)
211
+ concatenate_class = cv2.cvtColor(concatenate_class, cv2.COLOR_RGB2BGR)
212
 
213
  op.append((concatenate_class,features_id_neg))# intergrate negative features
214
  return op