fix weights

.gitignore

@@ -153,3 +153,4 @@ temp/
 **/*.glb
 **/*.bin
 data/
+**/*.pth

mvp.py

@@ -53,7 +53,16 @@ except TypeError:
 metric3d_model.to(device)
 metric3d_model.eval()
 
+cropformer_name = "Mask2Former_hornet_3x_576d0b.pth"
 
+def check_weights():
+    if not os.path.exists(os.path.join(MK_PATH, cropformer_name)):
+        print(f"Downloading {cropformer_name}...")
+        os.system(f"wget https://huggingface.co/datasets/qqlu1992/Adobe_EntitySeg/resolve/main/CropFormer_model/Entity_Segmentation/Mask2Former_hornet_3x/Mask2Former_hornet_3x_576d0b.pth?download=true -O {os.path.join(MK_PATH, cropformer_name)}")
+        print(f"Downloaded {cropformer_name}...")
+    else:
+        print(f"{cropformer_name} already exists...")
+check_weights()
 
 def load_clip():
     print(f'[INFO] loading CLIP model...')