Load sam inputs from S3

pytorch · Nov 25, 2024 · 615ac43 · 615ac43
1 parent 83fb47e
commit 615ac43
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/torchbenchmark/models/sam/__init__.py b/torchbenchmark/models/sam/__init__.py
@@ -35,7 +35,7 @@ def __init__(self, test, device, batch_size=1, extra_args=[]):
         self.model = sam_model_registry[model_type](checkpoint=sam_checkpoint)
         self.model.to(device=device)
         data_folder = os.path.dirname(DATA_PATH)
-        image_path = os.path.join(data_folder, "data", "sam", "truck.jpg")
+        image_path = os.path.join(data_folder, "data", ".data", "sam_inputs", "truck.jpg")
         if not os.path.exists(image_path):
             from torchbenchmark.util.framework.fb.installer import install_data
 

diff --git a/torchbenchmark/models/sam/install.py b/torchbenchmark/models/sam/install.py
@@ -27,7 +27,7 @@ def download_checkpoint():
 
 def download_data():
     s3_utils.checkout_s3_data(
-        "INPUT_TARBALLS", "sam.tar.gz", decompress=True
+        "INPUT_TARBALLS", "sam_inputs.tar.gz", decompress=True
     )
 
 

diff --git a/torchbenchmark/models/sam_fast/__init__.py b/torchbenchmark/models/sam_fast/__init__.py
@@ -31,7 +31,7 @@ def __init__(self, test, device, batch_size=1, extra_args=[]):
         self.model = sam_model_fast_registry[model_type](checkpoint=sam_checkpoint)
         self.model.to(device=device)
         data_folder = os.path.dirname(DATA_PATH)
-        image_path = os.path.join(data_folder, "data", "sam", "truck.jpg")
+        image_path = os.path.join(data_folder, "data", ".data", "sam_inputs", "truck.jpg")
         assert os.path.exists(image_path), f"Expected image file exists at {image_path} but not found."
         self.image = cv2.imread(image_path)
         self.image = cv2.cvtColor(self.image, cv2.COLOR_BGR2RGB)

diff --git a/torchbenchmark/models/sam_fast/install.py b/torchbenchmark/models/sam_fast/install.py
@@ -27,7 +27,7 @@ def download_checkpoint():
 
 def download_data():
     s3_utils.checkout_s3_data(
-        "INPUT_TARBALLS", "sam.tar.gz", decompress=True
+        "INPUT_TARBALLS", "sam_inputs.tar.gz", decompress=True
     )