hard target matching removed from check_mask_stats

inference_utils/output_processing.py

@@ -11,15 +11,15 @@ def check_mask_stats(img, mask, modality_type, target):
     # modality_type: str, see target_dist.json for the list of modality types
     # target: str, see target_dist.json for the list of targets
     
-    # huggingface_hub.hf_hub_download('microsoft/BiomedParse', filename='target_dist.json', local_dir='./inference_utils')
-    # huggingface_hub.hf_hub_download('microsoft/BiomedParse', filename="config.yaml", local_dir="./configs")
+    huggingface_hub.hf_hub_download('microsoft/BiomedParse', filename='target_dist.json', local_dir='./inference_utils')
+    huggingface_hub.hf_hub_download('microsoft/BiomedParse', filename="config.yaml", local_dir="./configs")
     target_dist = json.load(open("inference_utils/target_dist.json"))
     
-    # if modality_type not in target_dist:
-    #     raise ValueError(f"Currently support modality types: {list(target_dist.keys())}")
+    if modality_type not in target_dist:
+        raise ValueError(f"Currently support modality types: {list(target_dist.keys())}")
     
-    # if target not in target_dist[modality_type]:
-    #     raise ValueError(f"Currently support targets for {modality_type}: {list(target_dist[modality_type].keys())}")
+    if target not in target_dist[modality_type]:
+        raise ValueError(f"Currently support targets for {modality_type}: {list(target_dist[modality_type].keys())}")
     
     ms = mask_stats(mask, img)