Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix CI bug #809

Open
wants to merge 2 commits into
base: dev-1.x
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 2 additions & 3 deletions .dev_scripts/gather_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,14 +2,13 @@
import argparse
import glob
import json
import mmcv
import os.path as osp
import shutil
import subprocess
from collections import OrderedDict

import mmcv
import torch
import yaml
from collections import OrderedDict


def ordered_yaml_dump(data, stream=None, Dumper=yaml.SafeDumper, **kwds):
Expand Down
2 changes: 1 addition & 1 deletion .pre-commit-config-zh-cn.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ repos:
hooks:
- id: flake8
- repo: https://gitee.com/openmmlab/mirrors-isort
rev: 5.10.1
rev: 5.11.5
hooks:
- id: isort
- repo: https://gitee.com/openmmlab/mirrors-yapf
Expand Down
8 changes: 2 additions & 6 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -3,12 +3,8 @@ repos:
rev: 5.0.4
hooks:
- id: flake8
- repo: https://github.com/asottile/seed-isort-config
rev: v2.2.0
hooks:
- id: seed-isort-config
- repo: https://github.com/timothycrosley/isort
rev: 4.3.21
- repo: https://github.com/PyCQA/isort
rev: 5.11.5
hooks:
- id: isort
- repo: https://github.com/pre-commit/mirrors-yapf
Expand Down
3 changes: 1 addition & 2 deletions demo/huge_image_demo.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
# Copyright (c) OpenMMLab. All rights reserved.
from argparse import ArgumentParser

import mmcv
from argparse import ArgumentParser
from mmdet.apis import init_detector

from mmrotate.apis import inference_detector_by_patches
Expand Down
3 changes: 1 addition & 2 deletions demo/image_demo.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
# Copyright (c) OpenMMLab. All rights reserved.
from argparse import ArgumentParser

import mmcv
from argparse import ArgumentParser
from mmdet.apis import inference_detector, init_detector

from mmrotate.registry import VISUALIZERS
Expand Down
3 changes: 1 addition & 2 deletions docs/en/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,8 @@
# documentation root, use os.path.abspath to make it absolute, like shown here.
#
import os
import sys

import pytorch_sphinx_theme
import sys

sys.path.insert(0, os.path.abspath('../..'))

Expand Down
3 changes: 1 addition & 2 deletions docs/zh_cn/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,8 @@
# documentation root, use os.path.abspath to make it absolute, like shown here.
#
import os
import sys

import pytorch_sphinx_theme
import sys

sys.path.insert(0, os.path.abspath('../..'))

Expand Down
3 changes: 1 addition & 2 deletions mmrotate/apis/inference.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,12 @@
# Copyright (c) OpenMMLab. All rights reserved.
from typing import List, Optional, Sequence, Union

import mmcv
import numpy as np
import torch
from mmcv.ops import RoIPool
from mmcv.transforms import Compose
from mmdet.structures import DetDataSample, SampleList
from torch import nn
from typing import List, Optional, Sequence, Union

from mmrotate.utils import (get_multiscale_patch, get_test_pipeline_cfg,
merge_results_by_nms, slide_window)
Expand Down
7 changes: 3 additions & 4 deletions mmrotate/datasets/dior.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,11 @@
# Copyright (c) OpenMMLab. All rights reserved.
import os.path as osp
import xml.etree.ElementTree as ET
from typing import List, Optional, Union

import mmcv
import numpy as np
import os.path as osp
import xml.etree.ElementTree as ET
from mmengine.dataset import BaseDataset
from mmengine.fileio import get, get_local_path, list_from_file
from typing import List, Optional, Union

from mmrotate.registry import DATASETS

Expand Down
3 changes: 1 addition & 2 deletions mmrotate/datasets/dota.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,8 @@
# Copyright (c) OpenMMLab. All rights reserved.
import glob
import os.path as osp
from typing import List

from mmengine.dataset import BaseDataset
from typing import List

from mmrotate.registry import DATASETS

Expand Down
7 changes: 3 additions & 4 deletions mmrotate/datasets/hrsc.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,12 @@
# Copyright (c) OpenMMLab. All rights reserved.
import os.path as osp
import xml.etree.ElementTree as ET
from typing import List, Optional, Union

import mmcv
import numpy as np
import os.path as osp
import torch
import xml.etree.ElementTree as ET
from mmengine.dataset import BaseDataset
from mmengine.fileio import get, get_local_path, list_from_file
from typing import List, Optional, Union

from mmrotate.registry import DATASETS
from mmrotate.structures.bbox import rbox2qbox
Expand Down
3 changes: 1 addition & 2 deletions mmrotate/datasets/transforms/loading.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,7 @@
# Copyright (c) OpenMMLab. All rights reserved.
from typing import Sequence, Union

import mmcv
from mmcv.transforms import BaseTransform
from typing import Sequence, Union

from mmrotate.registry import TRANSFORMS

Expand Down
5 changes: 2 additions & 3 deletions mmrotate/datasets/transforms/transforms.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from numbers import Number
from typing import List, Optional, Union

import cv2
import mmcv
import numpy as np
Expand All @@ -10,6 +7,8 @@
from mmdet.structures.bbox import BaseBoxes, get_box_type
from mmdet.structures.mask import PolygonMasks
from mmengine.utils import is_list_of
from numbers import Number
from typing import List, Optional, Union

from mmrotate.registry import TRANSFORMS

Expand Down
3 changes: 1 addition & 2 deletions mmrotate/evaluation/functional/mean_ap.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,10 @@
# Copyright (c) OpenMMLab. All rights reserved.
from multiprocessing import get_context

import numpy as np
import torch
from mmcv.ops import box_iou_quadri, box_iou_rotated
from mmdet.evaluation.functional import average_precision
from mmengine.logging import print_log
from multiprocessing import get_context
from terminaltables import AsciiTable


Expand Down
7 changes: 3 additions & 4 deletions mmrotate/evaluation/metrics/dota_metric.py
Original file line number Diff line number Diff line change
@@ -1,19 +1,18 @@
# Copyright (c) OpenMMLab. All rights reserved.
import copy
import numpy as np
import os
import os.path as osp
import re
import tempfile
import torch
import zipfile
from collections import OrderedDict, defaultdict
from typing import List, Optional, Sequence, Union

import numpy as np
import torch
from mmcv.ops import nms_quadri, nms_rotated
from mmengine.evaluator import BaseMetric
from mmengine.fileio import dump
from mmengine.logging import MMLogger
from typing import List, Optional, Sequence, Union

from mmrotate.evaluation import eval_rbbox_map
from mmrotate.registry import METRICS
Expand Down
9 changes: 4 additions & 5 deletions mmrotate/evaluation/metrics/rotated_coco_metric.py
Original file line number Diff line number Diff line change
@@ -1,21 +1,20 @@
# Copyright (c) OpenMMLab. All rights reserved.
import cv2
import datetime
import itertools
import numpy as np
import os.path as osp
import pycocotools.mask as maskUtils
import tempfile
from collections import OrderedDict
from typing import Dict, Optional, Sequence

import cv2
import numpy as np
import pycocotools.mask as maskUtils
from mmcv.ops import box_iou_rotated
from mmdet.datasets.api_wrappers import COCO
from mmdet.evaluation import CocoMetric
from mmengine import MMLogger
from mmengine.fileio import dump, load
from pycocotools.cocoeval import COCOeval
from terminaltables import AsciiTable
from typing import Dict, Optional, Sequence

from mmrotate.registry import METRICS
from mmrotate.structures.bbox import RotatedBoxes
Expand Down
4 changes: 2 additions & 2 deletions mmrotate/models/backbones/re_resnet.py
Original file line number Diff line number Diff line change
@@ -1,19 +1,19 @@
# Copyright (c) OpenMMLab. All rights reserved.
# Modified from csuhan: https://github.com/csuhan/ReDet
from typing import Optional, Sequence, Tuple

import torch.nn as nn
import torch.utils.checkpoint as cp
from mmdet.utils import ConfigType, OptConfigType, OptMultiConfig
from mmengine.model import BaseModule
from torch import Tensor
from torch.nn.modules.batchnorm import _BatchNorm
from typing import Optional, Sequence, Tuple

from mmrotate.registry import MODELS

try:
import e2cnn.nn as enn
from e2cnn.nn import EquivariantModule

from ..utils.enn import (build_enn_divide_feature, build_enn_norm_layer,
build_enn_trivial_feature, ennAvgPool, ennConv,
ennMaxPool, ennReLU, ennTrivialConv)
Expand Down
3 changes: 1 addition & 2 deletions mmrotate/models/dense_heads/angle_branch_retina_head.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,5 @@
# Copyright (c) OpenMMLab. All rights reserved.
import copy
from typing import List, Optional, Tuple, Union

import torch
import torch.nn as nn
from mmdet.models.dense_heads.retina_head import RetinaHead
Expand All @@ -13,6 +11,7 @@
from mmengine.config import ConfigDict
from mmengine.structures import InstanceData
from torch import Tensor
from typing import List, Optional, Tuple, Union

from mmrotate.registry import MODELS, TASK_UTILS

Expand Down
3 changes: 1 addition & 2 deletions mmrotate/models/dense_heads/cfa_head.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,11 @@
# Copyright (c) OpenMMLab. All rights reserved.
from typing import Dict, List, Optional, Tuple

import numpy as np
import torch
from mmdet.models.utils import multi_apply
from mmdet.utils import InstanceList, OptInstanceList
from mmengine.structures import InstanceData
from torch import Tensor
from typing import Dict, List, Optional, Tuple

from mmrotate.models.dense_heads.rotated_reppoints_head import \
RotatedRepPointsHead
Expand Down
3 changes: 1 addition & 2 deletions mmrotate/models/dense_heads/h2rbox_head.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,6 @@
# Copyright (c) OpenMMLab. All rights reserved.
import copy
import math
from typing import Dict, List, Optional, Tuple

import torch
from mmcv.cnn import Scale
from mmdet.models.utils import (filter_scores_and_topk, multi_apply,
Expand All @@ -13,6 +11,7 @@
from mmengine import ConfigDict
from mmengine.structures import InstanceData
from torch import Tensor
from typing import Dict, List, Optional, Tuple

from mmrotate.models.dense_heads.rotated_fcos_head import RotatedFCOSHead
from mmrotate.registry import MODELS
Expand Down
3 changes: 1 addition & 2 deletions mmrotate/models/dense_heads/oriented_reppoints_head.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,5 @@
# Copyright (c) OpenMMLab. All rights reserved.
import math
from typing import Dict, List, Optional, Tuple

import numpy as np
import torch
import torch.nn as nn
Expand All @@ -11,6 +9,7 @@
from mmdet.utils import ConfigType, InstanceList, OptInstanceList
from mmengine.structures import InstanceData
from torch import Tensor
from typing import Dict, List, Optional, Tuple

from mmrotate.models.dense_heads.rotated_reppoints_head import \
RotatedRepPointsHead
Expand Down
3 changes: 1 addition & 2 deletions mmrotate/models/dense_heads/oriented_rpn_head.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,12 @@
# Copyright (c) OpenMMLab. All rights reserved.
from typing import Optional

import torch
from mmcv.ops import batched_nms
from mmdet.models.dense_heads import RPNHead
from mmdet.structures.bbox import (BaseBoxes, get_box_tensor, get_box_wh,
scale_boxes)
from mmengine.config import ConfigDict
from mmengine.structures import InstanceData
from typing import Optional

from mmrotate.registry import MODELS
from mmrotate.structures.bbox import rbox2hbox
Expand Down
3 changes: 1 addition & 2 deletions mmrotate/models/dense_heads/r3_head.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,10 @@
# Copyright (c) OpenMMLab. All rights reserved.
from typing import List, Optional, Tuple, Union

import torch
from mmdet.models.utils import select_single_mlvl
from mmdet.utils import InstanceList, OptInstanceList
from mmengine.config import ConfigDict
from torch import Tensor
from typing import List, Optional, Tuple, Union

from mmrotate.registry import MODELS
from mmrotate.structures.bbox import RotatedBoxes
Expand Down
3 changes: 1 addition & 2 deletions mmrotate/models/dense_heads/rotated_atss_head.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from typing import List, Optional

import torch
import torch.nn as nn
from mmcv.cnn import ConvModule, Scale
Expand All @@ -11,6 +9,7 @@
from mmdet.utils import InstanceList, OptInstanceList
from mmengine.structures import InstanceData
from torch import Tensor
from typing import List, Optional

from mmrotate.registry import MODELS
from mmrotate.structures.bbox import RotatedBoxes
Expand Down
3 changes: 1 addition & 2 deletions mmrotate/models/dense_heads/rotated_fcos_head.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,5 @@
# Copyright (c) OpenMMLab. All rights reserved.
import copy
from typing import Dict, List, Optional, Tuple

import torch
import torch.nn as nn
from mmcv.cnn import Scale
Expand All @@ -14,6 +12,7 @@
from mmengine import ConfigDict
from mmengine.structures import InstanceData
from torch import Tensor
from typing import Dict, List, Optional, Tuple

from mmrotate.registry import MODELS, TASK_UTILS
from mmrotate.structures import RotatedBoxes
Expand Down
3 changes: 1 addition & 2 deletions mmrotate/models/dense_heads/rotated_reppoints_head.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from typing import Dict, List, Tuple

import torch
from mmcv.ops import min_area_polygons
from mmdet.models.dense_heads.reppoints_head import RepPointsHead
Expand All @@ -9,6 +7,7 @@
from mmengine.config import ConfigDict
from mmengine.structures import InstanceData
from torch import Tensor
from typing import Dict, List, Tuple

from mmrotate.registry import MODELS
from mmrotate.structures.bbox import RotatedBoxes, qbox2rbox
Expand Down
Loading