allencbzhang commited on
Commit
8096d9e
·
verified ·
1 Parent(s): 3a26792

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -14
app.py CHANGED
@@ -7,15 +7,14 @@ import subprocess
7
  import shlex
8
  import torch
9
 
10
- os.system("pip install git+https://github.com/facebookresearch/detectron2.git")
11
- os.system("git clone https://github.com/Visual-AI/Mr.DETR.git MrDETR && cd MrDETR && rm -f requirements.txt && cd ..")
12
  subprocess.run(
13
  shlex.split(
14
  "pip install detrex-0.3.0-cp310-cp310-linux_x86_64.whl"
15
  )
16
  )
17
 
18
- sys.path.append("MrDETR/")
19
 
20
 
21
 
@@ -28,7 +27,7 @@ from PIL import Image
28
 
29
  if __name__ == "__main__":
30
  gr.close_all()
31
- cfg = LazyConfig.load("MrDETR/projects/mr_detr_align/configs/deformable_detr_swinl_two_stage_12ep_plusplus.py")
32
  cfg["model"].device = "cuda"
33
  cfg["train"].device = "cuda"
34
 
@@ -36,7 +35,7 @@ if __name__ == "__main__":
36
  # def
37
  model = instantiate(cfg.model)
38
  checkpointer = DetectionCheckpointer(model)
39
- checkpointer.load("https://github.com/Visual-AI/Mr.DETR/releases/download/weights/MrDETR_align_swinL_12ep_900q_safe.pth")
40
 
41
  model.eval()
42
  model.cuda()
@@ -64,16 +63,9 @@ if __name__ == "__main__":
64
  ],
65
  outputs="image",
66
  examples=[
67
- ["MrDETR/assets/000000014226.jpg", 0.5],
68
- ["MrDETR/assets/000000028449.jpg", 0.3],
69
- ["MrDETR/assets/000000070048.jpg", 0.5],
70
- ["MrDETR/assets/000000218997.jpg", 0.5],
71
- ["MrDETR/assets/000000279774.jpg", 0.5],
72
- ["MrDETR/assets/000000434459.jpg", 0.5],
73
- ["MrDETR/assets/000000448448.jpg", 0.5],
74
- ["MrDETR/assets/000000560474.jpg", 0.5],
75
  ],
76
- title="[CVPR 2025 highlight] ***v***-CLR: View-Consistent Learning for Open-World Instance Segmentation",
77
  description='''
78
  [![Paper](https://img.shields.io/badge/arXiv-2412.10028-red)](https://arxiv.org/abs/2412.10028)
79
  [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/mr-detr-instructive-multi-route-training-for/object-detection-on-coco-2017-val)](https://paperswithcode.com/sota/object-detection-on-coco-2017-val?p=mr-detr-instructive-multi-route-training-for)
 
7
  import shlex
8
  import torch
9
 
10
+ os.system("git clone https://github.com/Visual-AI/vCLR.git && cd vCLR && rm -f requirements.txt && cd ..")
 
11
  subprocess.run(
12
  shlex.split(
13
  "pip install detrex-0.3.0-cp310-cp310-linux_x86_64.whl"
14
  )
15
  )
16
 
17
+ sys.path.append("vCLR/")
18
 
19
 
20
 
 
27
 
28
  if __name__ == "__main__":
29
  gr.close_all()
30
+ cfg = LazyConfig.load("vCLR/projects/vCLR_deformable_mask/configs/dino-resnet/deformable_train_voc_eval_nonvoc.py")
31
  cfg["model"].device = "cuda"
32
  cfg["train"].device = "cuda"
33
 
 
35
  # def
36
  model = instantiate(cfg.model)
37
  checkpointer = DetectionCheckpointer(model)
38
+ checkpointer.load("https://huggingface.co/allencbzhang/vCLR/resolve/main/vCLR_deformable_train_on_voc.pth")
39
 
40
  model.eval()
41
  model.cuda()
 
63
  ],
64
  outputs="image",
65
  examples=[
66
+
 
 
 
 
 
 
 
67
  ],
68
+ title="[CVPR 2025 highlight] v-CLR: View-Consistent Learning for Open-World Instance Segmentation",
69
  description='''
70
  [![Paper](https://img.shields.io/badge/arXiv-2412.10028-red)](https://arxiv.org/abs/2412.10028)
71
  [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/mr-detr-instructive-multi-route-training-for/object-detection-on-coco-2017-val)](https://paperswithcode.com/sota/object-detection-on-coco-2017-val?p=mr-detr-instructive-multi-route-training-for)