fix data processing bugs

pkuhxy · Dec 24, 2024 · 21a87f6 · 21a87f6
1 parent 42814d9
commit 21a87f6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/data_preprocess/step1_get_bbox_pose.py b/data_preprocess/step1_get_bbox_pose.py
@@ -7,7 +7,7 @@
 from huggingface_hub import hf_hub_download
 from tqdm import tqdm
 from ultralytics import YOLO
-from util.download_weights import download_file
+from util.download_weights_data import download_file
 
 
 def parse_args():

diff --git a/data_preprocess/step4_get_mask.py b/data_preprocess/step4_get_mask.py
@@ -363,6 +363,6 @@ def main(json_folder, video_folder, output_path, sam2_checkpoint, model_cfg):
         print("Model not found, downloading from Hugging Face and Github...")
         hf_hub_download(repo_id="facebook/sam2.1-hiera-large", filename="sam2.1_hiera_large.pt", local_dir=args.sam2_checkpoint_path)
     else:
-        print(f"Model already exists in {args.model_path}, skipping download.")
+        print(f"Model already exists in {args.sam2_checkpoint_path}, skipping download.")
 
     main(args.json_folder, args.video_folder, args.output_path, sam2_checkpoint, args.model_cfg)