wendi.hwd yingda.chen 3 years ago
parent
commit
f4044f14fd
19 changed files with 36 additions and 9 deletions
  1. +1
    -0
      modelscope/models/cv/object_detection/mmdet_model.py
  2. +2
    -0
      modelscope/models/cv/object_detection/mmdet_ms/__init__.py
  3. +2
    -0
      modelscope/models/cv/object_detection/mmdet_ms/backbones/__init__.py
  4. +2
    -0
      modelscope/models/cv/object_detection/mmdet_ms/dense_heads/__init__.py
  5. +2
    -1
      modelscope/models/cv/object_detection/mmdet_ms/dense_heads/anchor_head.py
  6. +2
    -1
      modelscope/models/cv/object_detection/mmdet_ms/dense_heads/rpn_head.py
  7. +2
    -0
      modelscope/models/cv/object_detection/mmdet_ms/necks/__init__.py
  8. +2
    -1
      modelscope/models/cv/object_detection/mmdet_ms/necks/fpn.py
  9. +2
    -0
      modelscope/models/cv/object_detection/mmdet_ms/roi_heads/__init__.py
  10. +2
    -0
      modelscope/models/cv/object_detection/mmdet_ms/roi_heads/bbox_heads/__init__.py
  11. +2
    -1
      modelscope/models/cv/object_detection/mmdet_ms/roi_heads/bbox_heads/convfc_bbox_head.py
  12. +2
    -0
      modelscope/models/cv/object_detection/mmdet_ms/roi_heads/mask_heads/__init__.py
  13. +2
    -1
      modelscope/models/cv/object_detection/mmdet_ms/roi_heads/mask_heads/fcn_mask_head.py
  14. +2
    -0
      modelscope/models/cv/object_detection/mmdet_ms/utils/__init__.py
  15. +2
    -1
      modelscope/models/cv/object_detection/mmdet_ms/utils/checkpoint.py
  16. +2
    -2
      modelscope/models/cv/object_detection/mmdet_ms/utils/convModule_norm.py
  17. +2
    -0
      modelscope/models/cv/salient_detection/models/__init__.py
  18. +2
    -1
      modelscope/models/cv/salient_detection/models/u2net.py
  19. +1
    -0
      modelscope/models/cv/salient_detection/salient_model.py

+ 1
- 0
modelscope/models/cv/object_detection/mmdet_model.py View File

@@ -1,3 +1,4 @@
# Copyright (c) Alibaba, Inc. and its affiliates.
import os.path as osp import os.path as osp


import numpy as np import numpy as np


+ 2
- 0
modelscope/models/cv/object_detection/mmdet_ms/__init__.py View File

@@ -1,3 +1,5 @@
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
from .backbones import ViT from .backbones import ViT
from .dense_heads import AnchorNHead, RPNNHead from .dense_heads import AnchorNHead, RPNNHead
from .necks import FPNF from .necks import FPNF


+ 2
- 0
modelscope/models/cv/object_detection/mmdet_ms/backbones/__init__.py View File

@@ -1,3 +1,5 @@
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
from .vit import ViT from .vit import ViT


__all__ = ['ViT'] __all__ = ['ViT']

+ 2
- 0
modelscope/models/cv/object_detection/mmdet_ms/dense_heads/__init__.py View File

@@ -1,3 +1,5 @@
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
from .anchor_head import AnchorNHead from .anchor_head import AnchorNHead
from .rpn_head import RPNNHead from .rpn_head import RPNNHead




+ 2
- 1
modelscope/models/cv/object_detection/mmdet_ms/dense_heads/anchor_head.py View File

@@ -1,5 +1,6 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
# Implementation in this file is modifed from source code avaiable via https://github.com/ViTAE-Transformer/ViTDet
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
from mmdet.models.builder import HEADS from mmdet.models.builder import HEADS
from mmdet.models.dense_heads import AnchorHead from mmdet.models.dense_heads import AnchorHead




+ 2
- 1
modelscope/models/cv/object_detection/mmdet_ms/dense_heads/rpn_head.py View File

@@ -1,5 +1,6 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
# Implementation in this file is modifed from source code avaiable via https://github.com/ViTAE-Transformer/ViTDet
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
import copy import copy


import torch import torch


+ 2
- 0
modelscope/models/cv/object_detection/mmdet_ms/necks/__init__.py View File

@@ -1,3 +1,5 @@
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
from .fpn import FPNF from .fpn import FPNF


__all__ = ['FPNF'] __all__ = ['FPNF']

+ 2
- 1
modelscope/models/cv/object_detection/mmdet_ms/necks/fpn.py View File

@@ -1,5 +1,6 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
# Implementation in this file is modifed from source code avaiable via https://github.com/ViTAE-Transformer/ViTDet
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from mmcv.runner import BaseModule, auto_fp16 from mmcv.runner import BaseModule, auto_fp16


+ 2
- 0
modelscope/models/cv/object_detection/mmdet_ms/roi_heads/__init__.py View File

@@ -1,3 +1,5 @@
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
from .bbox_heads import (ConvFCBBoxNHead, Shared2FCBBoxNHead, from .bbox_heads import (ConvFCBBoxNHead, Shared2FCBBoxNHead,
Shared4Conv1FCBBoxNHead) Shared4Conv1FCBBoxNHead)
from .mask_heads import FCNMaskNHead from .mask_heads import FCNMaskNHead


+ 2
- 0
modelscope/models/cv/object_detection/mmdet_ms/roi_heads/bbox_heads/__init__.py View File

@@ -1,3 +1,5 @@
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
from .convfc_bbox_head import (ConvFCBBoxNHead, Shared2FCBBoxNHead, from .convfc_bbox_head import (ConvFCBBoxNHead, Shared2FCBBoxNHead,
Shared4Conv1FCBBoxNHead) Shared4Conv1FCBBoxNHead)




+ 2
- 1
modelscope/models/cv/object_detection/mmdet_ms/roi_heads/bbox_heads/convfc_bbox_head.py View File

@@ -1,5 +1,6 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
# Implementation in this file is modifed from source code avaiable via https://github.com/ViTAE-Transformer/ViTDet
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
import torch.nn as nn import torch.nn as nn
from mmdet.models.builder import HEADS from mmdet.models.builder import HEADS
from mmdet.models.roi_heads.bbox_heads.bbox_head import BBoxHead from mmdet.models.roi_heads.bbox_heads.bbox_head import BBoxHead


+ 2
- 0
modelscope/models/cv/object_detection/mmdet_ms/roi_heads/mask_heads/__init__.py View File

@@ -1,3 +1,5 @@
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
from .fcn_mask_head import FCNMaskNHead from .fcn_mask_head import FCNMaskNHead


__all__ = ['FCNMaskNHead'] __all__ = ['FCNMaskNHead']

+ 2
- 1
modelscope/models/cv/object_detection/mmdet_ms/roi_heads/mask_heads/fcn_mask_head.py View File

@@ -1,5 +1,6 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
# Implementation in this file is modifed from source code avaiable via https://github.com/ViTAE-Transformer/ViTDet
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
from warnings import warn from warnings import warn


import numpy as np import numpy as np


+ 2
- 0
modelscope/models/cv/object_detection/mmdet_ms/utils/__init__.py View File

@@ -1,3 +1,5 @@
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
from .checkpoint import load_checkpoint from .checkpoint import load_checkpoint
from .convModule_norm import ConvModule_Norm from .convModule_norm import ConvModule_Norm




+ 2
- 1
modelscope/models/cv/object_detection/mmdet_ms/utils/checkpoint.py View File

@@ -1,5 +1,6 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) Open-MMLab. All rights reserved.
# Implementation adopted from ViTAE-Transformer, source code avaiable via https://github.com/ViTAE-Transformer/ViTDet
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
import io import io
import os import os
import os.path as osp import os.path as osp


+ 2
- 2
modelscope/models/cv/object_detection/mmdet_ms/utils/convModule_norm.py View File

@@ -1,5 +1,5 @@
# Implementation adopted from ViTAE-Transformer, source code avaiable via https://github.com/ViTAE-Transformer/ViTDet
# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly avaialbe at https://github.com/ViTAE-Transformer/ViTDet
from mmcv.cnn import ConvModule from mmcv.cnn import ConvModule






+ 2
- 0
modelscope/models/cv/salient_detection/models/__init__.py View File

@@ -1 +1,3 @@
# The implementation is adopted from U-2-Net, made publicly available under the Apache 2.0 License
# source code avaiable via https://github.com/xuebinqin/U-2-Net
from .u2net import U2NET from .u2net import U2NET

+ 2
- 1
modelscope/models/cv/salient_detection/models/u2net.py View File

@@ -1,4 +1,5 @@
# Implementation in this file is modifed from source code avaiable via https://github.com/xuebinqin/U-2-Net
# The implementation is adopted from U-2-Net, made publicly available under the Apache 2.0 License
# source code avaiable via https://github.com/xuebinqin/U-2-Net
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F


+ 1
- 0
modelscope/models/cv/salient_detection/salient_model.py View File

@@ -1,3 +1,4 @@
# Copyright (c) Alibaba, Inc. and its affiliates.
import os.path as osp import os.path as osp


import cv2 import cv2


Loading…
Cancel
Save