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 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 .dense_heads import AnchorNHead, RPNNHead
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

__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 .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.
# 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.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.
# 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 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

__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.
# 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.functional as F
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,
Shared4Conv1FCBBoxNHead)
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,
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.
# 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
from mmdet.models.builder import HEADS
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

__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.
# 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

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 .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.
# 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 os
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




+ 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

+ 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.nn as nn
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 cv2


Loading…
Cancel
Save