sharvari0b26 commited on
Commit
f6526ef
·
1 Parent(s): b915ef3

Add changes to script.py

Browse files
Files changed (1) hide show
  1. script.py +1 -0
script.py CHANGED
@@ -14,6 +14,7 @@ def run_inference(TEST_IMAGE_PATH, pipeline_model, SUBMISSION_CSV_SAVE_PATH):
14
  for test_image in test_images:
15
  path_to_image = os.path.join(TEST_IMAGE_PATH, test_image)
16
  image = cv2.imread(path_to_image)
 
17
  features = extract_features_from_image(image)
18
  image_feature_list.append(features)
19
 
 
14
  for test_image in test_images:
15
  path_to_image = os.path.join(TEST_IMAGE_PATH, test_image)
16
  image = cv2.imread(path_to_image)
17
+ image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
18
  features = extract_features_from_image(image)
19
  image_feature_list.append(features)
20