From 21a87f6c266421b3e878ab9d209370c66fef992b Mon Sep 17 00:00:00 2001 From: DuNGEOnmassster Date: Tue, 24 Dec 2024 03:47:39 +0000 Subject: [PATCH] fix data processing bugs --- data_preprocess/step1_get_bbox_pose.py | 2 +- data_preprocess/step4_get_mask.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/data_preprocess/step1_get_bbox_pose.py b/data_preprocess/step1_get_bbox_pose.py index 8522ed3..7d444f0 100755 --- 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 index f6d5fb8..bd90606 100755 --- 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)