Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -3,7 +3,7 @@ image_extensions.add('webp')
|
|
| 3 |
|
| 4 |
class MultiCaReClassifier():
|
| 5 |
|
| 6 |
-
def __init__(self, image_folder, models_root, save_path = '', add_multiclass_columns = False):
|
| 7 |
|
| 8 |
'''Class used to classify medical images considering their types (such as ultrasound or MRI), and the corresponding anatomical region and view (for radiology images only).
|
| 9 |
image_folder (str): folder containing all the input images.
|
|
|
|
| 3 |
|
| 4 |
class MultiCaReClassifier():
|
| 5 |
|
| 6 |
+
def __init__(self, image_folder, models_root = 'MultiCaReClassifier/models', save_path = '', add_multiclass_columns = False):
|
| 7 |
|
| 8 |
'''Class used to classify medical images considering their types (such as ultrasound or MRI), and the corresponding anatomical region and view (for radiology images only).
|
| 9 |
image_folder (str): folder containing all the input images.
|