update quick demo code

Author: FanChiMao

assets/download_demo_images.py

@@ -1,22 +1,28 @@
 import os
 import wget
+import shutil
 import zipfile
+import argparse
 
-def main():
-    wget.download(
-        'https://github.com/FanChiMao/Competition-2024-PyTorch-Tracking/releases/download/v0.0/demo_images.zip')
+def main(target_path):
+    target_path = os.path.abspath(target_path)
+    wget.download('https://github.com/FanChiMao/Competition-2024-PyTorch-Tracking/releases/download/v0.0/demo_images.zip',
+                  out=target_path)
 
     try:
         print("Start download the demo images...")
-        if os.path.exists("./demo_images.zip"):
-            with zipfile.ZipFile("demo_images.zip") as zip_ref:
-                zip_ref.extractall(".")
+        if os.path.exists(os.path.join(target_path, "./demo_images.zip")):
+            with zipfile.ZipFile(os.path.join(target_path, "./demo_images.zip")) as zip_ref:
+                zip_ref.extractall(target_path)
     except Exception as error:
         print(f"Error Exception: {error}")
     finally:
-        if os.path.exists("./demo_images.zip"):
-            os.remove("./demo_images.zip")
+        if os.path.exists(os.path.join(target_path, "./demo_images.zip")):
+            os.remove(os.path.join(target_path, "./demo_images.zip"))
 
 
 if __name__ == '__main__':
-    main()
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--target_path', type=str, default=".")
+    args = parser.parse_args()
+    main(args.target_path)

demo_examples.py

@@ -1,6 +1,4 @@
 import os
-import sys
-sys.path.append("Detector")
 
 print(">> Download pretrained weights")
 try:
@@ -15,7 +13,7 @@
     os.makedirs("assets", exist_ok=True)
 if not os.path.exists("assets/demo_images"):
     try:
-        os.system("python assets/download_demo_images.py")
+        os.system("python assets/download_demo_images.py --target_path assets")
         if not os.path.exists("assets/demo_images"):
             raise "demo_images doesn't exist"
     except Exception as error: