# -*- coding: utf-8 -*-
# @Author  : LG

model_dict = {'sam_vit_h_4b8939.pth':
                  {
                      'urls': ['https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth',
                               'https://www.modelscope.cn/api/v1/models/yatengLG/ISAT_with_segment_anything_checkpoints/repo?Revision=master&FilePath=checkpoints/sam_vit_h_4b8939.pth'],
                      'memory':'6462M',
                      'bf16_memory':'3378M',
                      'params':'2.6G'
                  },
              'sam_vit_l_0b3195.pth':
                  {
                      'urls': ['https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth',
                               'https://www.modelscope.cn/api/v1/models/yatengLG/ISAT_with_segment_anything_checkpoints/repo?Revision=master&FilePath=checkpoints/sam_vit_l_0b3195.pth'],
                      'memory': '5016M',
                      'bf16_memory': '2634M',
                      'params': '1.3G'
                   },
              'sam_vit_b_01ec64.pth':
                  {
                      'urls': ['https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth',
                              'https://www.modelscope.cn/api/v1/models/yatengLG/ISAT_with_segment_anything_checkpoints/repo?Revision=master&FilePath=checkpoints/sam_vit_b_01ec64.pth'],
                      'memory': '3302M',
                      'bf16_memory': '1760M',
                      'params': '375M'
                  },
              'sam_hq_vit_h.pth':
                  {
                      'urls': ['https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_h.pth',
                               'https://www.modelscope.cn/api/v1/models/yatengLG/ISAT_with_segment_anything_checkpoints/repo?Revision=master&FilePath=checkpoints/sam_hq_vit_h.pth'],
                      'memory': '6464M',
                      'bf16_memory': '3378M',
                      'params': '2.6G'
                  },
              'sam_hq_vit_l.pth':
                  {
                      'urls': ['https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_l.pth',
                               'https://www.modelscope.cn/api/v1/models/yatengLG/ISAT_with_segment_anything_checkpoints/repo?Revision=master&FilePath=checkpoints/sam_hq_vit_l.pth'],
                      'memory': '5016M',
                      'bf16_memory': '2634M',
                      'params': '1.3G'
                  },
              'sam_hq_vit_b.pth':
                  {
                      'urls': ['https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_b.pth',
                               'https://www.modelscope.cn/api/v1/models/yatengLG/ISAT_with_segment_anything_checkpoints/repo?Revision=master&FilePath=checkpoints/sam_hq_vit_b.pth'],
                      'memory': '3304M',
                      'bf16_memory': '1762M',
                      'params': '379M'
                   },
              'sam_hq_vit_tiny.pth':
                  {
                      'urls': ['https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_tiny.pth',
                              'https://www.modelscope.cn/api/v1/models/yatengLG/ISAT_with_segment_anything_checkpoints/repo?Revision=master&FilePath=checkpoints/sam_hq_vit_tiny.pth'],
                      'memory': '598M',
                      'bf16_memory': '392M',
                      'params': '43M'
                  },
              'mobile_sam.pt':
                  {
                      'urls': ['https://huggingface.co/yateng/isat_checkpoints/resolve/main/mobile_sam.pt',
                               'https://www.modelscope.cn/api/v1/models/yatengLG/ISAT_with_segment_anything_checkpoints/repo?Revision=master&FilePath=checkpoints/mobile_sam.pt'],
                      'memory': '534M',
                      'bf16_memory': '390M',
                      'params': '40M'
                  },
              'edge_sam.pth':
                  {
                      'urls': ['https://huggingface.co/spaces/chongzhou/EdgeSAM/resolve/main/weights/edge_sam.pth',
                               'https://www.modelscope.cn/api/v1/models/yatengLG/ISAT_with_segment_anything_checkpoints/repo?Revision=master&FilePath=checkpoints/edge_sam.pth'],
                      'memory': '360M',
                      'bf16_memory': '304M',
                      'params': '38.8M'
                  },
              'edge_sam_3x.pth':
                  {
                      'urls': ['https://huggingface.co/spaces/chongzhou/EdgeSAM/resolve/main/weights/edge_sam_3x.pth',
                               'https://www.modelscope.cn/api/v1/models/yatengLG/ISAT_with_segment_anything_checkpoints/repo?Revision=master&FilePath=checkpoints/edge_sam_3x.pth'],
                      'memory': '360M',
                      'bf16_memory': '304M',
                      'params': '38.8M'
                  },
    'sam-med2d_b.pth':
                  {
                      'urls': ['https://huggingface.co/yatengLG/ISAT_with_segment_anything_checkpoints/resolve/main/sam-med2d_b.pth',
                               'https://www.modelscope.cn/api/v1/models/yatengLG/ISAT_with_segment_anything_checkpoints/repo?Revision=master&FilePath=checkpoints/sam-med2d_b.pth'],
                      'memory': '1500M',
                      'bf16_memory': '1050M',
                      'params': '2.4G'
                  },
              }
