Update visualization_gary.py
Browse files- 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
|