Commit ·
f6526ef
1
Parent(s): b915ef3
Add changes to script.py
Browse files
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 |
|