rockeycoss commited on
Commit
073441b
·
1 Parent(s): a6190ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -8,7 +8,10 @@ import torch
8
  # torch_ver, cuda_ver = torch.__version__.split('+')
9
  # os.system('pip list')
10
  # os.system(f'pip install pycocotools==2.0.0 mmdet mmcv-full==1.5.0 -f https://download.openmmlab.com/mmcv/dist/{cuda_ver}/torch1.10.0/index.html --no-cache-dir')
11
-
 
 
 
12
 
13
  from mmcv import Config
14
  from mmcv.utils import IS_CUDA_AVAILABLE, IS_MLU_AVAILABLE
 
8
  # torch_ver, cuda_ver = torch.__version__.split('+')
9
  # os.system('pip list')
10
  # os.system(f'pip install pycocotools==2.0.0 mmdet mmcv-full==1.5.0 -f https://download.openmmlab.com/mmcv/dist/{cuda_ver}/torch1.10.0/index.html --no-cache-dir')
11
+ os.system(r'python -m wget https://github.com/HDETR/H-Deformable-DETR/releases/download/v0.1/r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_36eps.pth -o ckpt/r50_hdetr.pth')
12
+ os.system(r'python -m wget https://github.com/HDETR/H-Deformable-DETR/releases/download/v0.1/swin_tiny_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_36eps.pth -o ckpt/swin_t_hdetr.pth')
13
+ os.system(r'python tools/convert_ckpt.py ckpt/r50_hdetr.pth ckpt/r50_hdetr.pth')
14
+ os.system(r'python tools/convert_ckpt.py ckpt/swin_t_hdetr.pth ckpt/swin_t_hdetr.pth')
15
 
16
  from mmcv import Config
17
  from mmcv.utils import IS_CUDA_AVAILABLE, IS_MLU_AVAILABLE