fix: updating integ tests to use postProcessing parameter in image re…

…quests
aws-solutions-library-samples · Jul 17, 2024 · cfab852 · cfab852
1 parent 2c78ddf
commit cfab852
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/aws/osml/utils/integ_utils.py b/src/aws/osml/utils/integ_utils.py
@@ -440,7 +440,7 @@ def build_image_processing_request(endpoint: str, endpoint_type: str, image_url:
         "imageProcessorTileOverlap": OSMLConfig.TILE_OVERLAP,
         "imageProcessorTileFormat": OSMLConfig.TILE_FORMAT,
         "imageProcessorTileCompression": OSMLConfig.TILE_COMPRESSION,
-        "featureSelectionOptions": OSMLConfig.FEATURE_SELECTION_OPTIONS,
+        "postProcessing": json.loads(OSMLConfig.POST_PROCESSING),
         "regionOfInterest": OSMLConfig.REGION_OF_INTEREST,
     }
 

diff --git a/src/aws/osml/utils/osml_config.py b/src/aws/osml/utils/osml_config.py
@@ -51,9 +51,9 @@ class OSMLConfig:
     TILE_COMPRESSION: str = os.environ.get("TILE_COMPRESSION", "NONE")
     TILE_SIZE: int = int(os.environ.get("TILE_SIZE", "512"))
     TILE_OVERLAP: int = int(os.environ.get("TILE_OVERLAP", "128"))
-    FEATURE_SELECTION_OPTIONS: str = os.environ.get(
-        "FEATURE_SELECTION_OPTIONS",
-        '{"algorithm": "NMS", "iou_threshold":  0.75, "skip_box_threshold": 0.0001, "sigma": .1}',
+    POST_PROCESSING: str = os.environ.get(
+        "POST_PROCESSING",
+        '[{"step": "FEATURE_DISTILLATION", "algorithm": {"algorithmType": "NMS", "iouThreshold":  0.75}}]',
     )
     REGION_OF_INTEREST: str = os.environ.get("REGION_OF_INTEREST")