At first, change default value of erase_color parameter to 0.
|
def feature_extraction_segment( |
|
img: np.ndarray, |
|
segments: np.ndarray, |
|
label: int, |
|
max_radius: Optional[int] = None, |
|
centroid: Optional[List[int]] = None, |
|
erase_color: Optional[int] = None, |
|
feature_method: FeaturesMethods = "comatrix" |
|
): |
After running the benchmark, if metrics was improved by this change, consider erase_color being some kind of hyperparameter for EGSIS.