Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
SOLOv2-pytorch
Commits
d2483e15
Unverified
Commit
d2483e15
authored
Jul 31, 2019
by
Kai Chen
Committed by
GitHub
Jul 31, 2019
Browse files
Use isort to sort imports and setup travis (#1085)
* add isort config * use isort to sort imports * add isort to travis
parent
864880de
Changes
65
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
38 additions
and
42 deletions
+38
-42
mmdet/models/anchor_heads/ga_retina_head.py
mmdet/models/anchor_heads/ga_retina_head.py
+3
-3
mmdet/models/anchor_heads/ga_rpn_head.py
mmdet/models/anchor_heads/ga_rpn_head.py
+1
-1
mmdet/models/anchor_heads/guided_anchor_head.py
mmdet/models/anchor_heads/guided_anchor_head.py
+4
-4
mmdet/models/anchor_heads/retina_head.py
mmdet/models/anchor_heads/retina_head.py
+2
-2
mmdet/models/anchor_heads/rpn_head.py
mmdet/models/anchor_heads/rpn_head.py
+1
-1
mmdet/models/anchor_heads/ssd_head.py
mmdet/models/anchor_heads/ssd_head.py
+1
-1
mmdet/models/backbones/hrnet.py
mmdet/models/backbones/hrnet.py
+1
-1
mmdet/models/backbones/resnet.py
mmdet/models/backbones/resnet.py
+2
-4
mmdet/models/backbones/resnext.py
mmdet/models/backbones/resnext.py
+2
-2
mmdet/models/backbones/ssd_vgg.py
mmdet/models/backbones/ssd_vgg.py
+1
-2
mmdet/models/bbox_heads/bbox_head.py
mmdet/models/bbox_heads/bbox_head.py
+2
-2
mmdet/models/bbox_heads/convfc_bbox_head.py
mmdet/models/bbox_heads/convfc_bbox_head.py
+1
-1
mmdet/models/builder.py
mmdet/models/builder.py
+2
-2
mmdet/models/detectors/base.py
mmdet/models/detectors/base.py
+2
-2
mmdet/models/detectors/cascade_rcnn.py
mmdet/models/detectors/cascade_rcnn.py
+4
-4
mmdet/models/detectors/fast_rcnn.py
mmdet/models/detectors/fast_rcnn.py
+1
-1
mmdet/models/detectors/faster_rcnn.py
mmdet/models/detectors/faster_rcnn.py
+1
-1
mmdet/models/detectors/fcos.py
mmdet/models/detectors/fcos.py
+1
-1
mmdet/models/detectors/grid_rcnn.py
mmdet/models/detectors/grid_rcnn.py
+3
-4
mmdet/models/detectors/htc.py
mmdet/models/detectors/htc.py
+3
-3
No files found.
mmdet/models/anchor_heads/ga_retina_head.py
View file @
d2483e15
import
torch.nn
as
nn
from
mmcv.cnn
import
normal_init
from
.guided_anchor_head
import
GuidedAnchorHead
,
FeatureAdaption
from
..registry
import
HEADS
from
..utils
import
bias_init_with_prob
,
ConvModule
from
mmdet.ops
import
MaskedConv2d
from
..registry
import
HEADS
from
..utils
import
ConvModule
,
bias_init_with_prob
from
.guided_anchor_head
import
FeatureAdaption
,
GuidedAnchorHead
@
HEADS
.
register_module
...
...
mmdet/models/anchor_heads/ga_rpn_head.py
View file @
d2483e15
...
...
@@ -5,8 +5,8 @@ from mmcv.cnn import normal_init
from
mmdet.core
import
delta2bbox
from
mmdet.ops
import
nms
from
.guided_anchor_head
import
GuidedAnchorHead
from
..registry
import
HEADS
from
.guided_anchor_head
import
GuidedAnchorHead
@
HEADS
.
register_module
...
...
mmdet/models/anchor_heads/guided_anchor_head.py
View file @
d2483e15
...
...
@@ -5,14 +5,14 @@ import torch
import
torch.nn
as
nn
from
mmcv.cnn
import
normal_init
from
mmdet.core
import
(
AnchorGenerator
,
anchor_
target
,
anchor_inside_flags
,
ga_loc_target
,
ga_shape_target
,
delta2bbox
,
multi_apply
,
multiclass_nms
,
force_fp32
)
from
mmdet.core
import
(
AnchorGenerator
,
anchor_
inside_flags
,
anchor_target
,
delta2bbox
,
force_fp32
,
ga_loc_target
,
ga_shape_target
,
multi_apply
,
multiclass_nms
)
from
mmdet.ops
import
DeformConv
,
MaskedConv2d
from
..builder
import
build_loss
from
.anchor_head
import
AnchorHead
from
..registry
import
HEADS
from
..utils
import
bias_init_with_prob
from
.anchor_head
import
AnchorHead
class
FeatureAdaption
(
nn
.
Module
):
...
...
mmdet/models/anchor_heads/retina_head.py
View file @
d2483e15
...
...
@@ -2,9 +2,9 @@ import numpy as np
import
torch.nn
as
nn
from
mmcv.cnn
import
normal_init
from
.anchor_head
import
AnchorHead
from
..registry
import
HEADS
from
..utils
import
bias_init_with_prob
,
ConvModule
from
..utils
import
ConvModule
,
bias_init_with_prob
from
.anchor_head
import
AnchorHead
@
HEADS
.
register_module
...
...
mmdet/models/anchor_heads/rpn_head.py
View file @
d2483e15
...
...
@@ -5,8 +5,8 @@ from mmcv.cnn import normal_init
from
mmdet.core
import
delta2bbox
from
mmdet.ops
import
nms
from
.anchor_head
import
AnchorHead
from
..registry
import
HEADS
from
.anchor_head
import
AnchorHead
@
HEADS
.
register_module
...
...
mmdet/models/anchor_heads/ssd_head.py
View file @
d2483e15
...
...
@@ -5,9 +5,9 @@ import torch.nn.functional as F
from
mmcv.cnn
import
xavier_init
from
mmdet.core
import
AnchorGenerator
,
anchor_target
,
multi_apply
from
.anchor_head
import
AnchorHead
from
..losses
import
smooth_l1_loss
from
..registry
import
HEADS
from
.anchor_head
import
AnchorHead
# TODO: add loss evaluator for SSD
...
...
mmdet/models/backbones/hrnet.py
View file @
d2483e15
...
...
@@ -6,7 +6,7 @@ from mmcv.runner import load_checkpoint
from
torch.nn.modules.batchnorm
import
_BatchNorm
from
..registry
import
BACKBONES
from
..utils
import
build_
norm
_layer
,
build_
conv
_layer
from
..utils
import
build_
conv
_layer
,
build_
norm
_layer
from
.resnet
import
BasicBlock
,
Bottleneck
...
...
mmdet/models/backbones/resnet.py
View file @
d2483e15
...
...
@@ -2,14 +2,12 @@ import logging
import
torch.nn
as
nn
import
torch.utils.checkpoint
as
cp
from
torch.nn.modules.batchnorm
import
_BatchNorm
from
mmcv.cnn
import
constant_init
,
kaiming_init
from
mmcv.runner
import
load_checkpoint
from
torch.nn.modules.batchnorm
import
_BatchNorm
from
mmdet.ops
import
DeformConv
,
ModulatedDeformConv
,
ContextBlock
from
mmdet.models.plugins
import
GeneralizedAttention
from
mmdet.ops
import
ContextBlock
,
DeformConv
,
ModulatedDeformConv
from
..registry
import
BACKBONES
from
..utils
import
build_conv_layer
,
build_norm_layer
...
...
mmdet/models/backbones/resnext.py
View file @
d2483e15
...
...
@@ -3,10 +3,10 @@ import math
import
torch.nn
as
nn
from
mmdet.ops
import
DeformConv
,
ModulatedDeformConv
from
.resnet
import
Bottleneck
as
_Bottleneck
from
.resnet
import
ResNet
from
..registry
import
BACKBONES
from
..utils
import
build_conv_layer
,
build_norm_layer
from
.resnet
import
Bottleneck
as
_Bottleneck
from
.resnet
import
ResNet
class
Bottleneck
(
_Bottleneck
):
...
...
mmdet/models/backbones/ssd_vgg.py
View file @
d2483e15
...
...
@@ -3,8 +3,7 @@ import logging
import
torch
import
torch.nn
as
nn
import
torch.nn.functional
as
F
from
mmcv.cnn
import
(
VGG
,
xavier_init
,
constant_init
,
kaiming_init
,
normal_init
)
from
mmcv.cnn
import
VGG
,
constant_init
,
kaiming_init
,
normal_init
,
xavier_init
from
mmcv.runner
import
load_checkpoint
from
..registry
import
BACKBONES
...
...
mmdet/models/bbox_heads/bbox_head.py
View file @
d2483e15
...
...
@@ -2,8 +2,8 @@ import torch
import
torch.nn
as
nn
import
torch.nn.functional
as
F
from
mmdet.core
import
(
delta2bbox
,
multiclass_nms
,
bbox_target
,
force_fp32
,
auto_fp16
)
from
mmdet.core
import
(
auto_fp16
,
bbox_target
,
delta2bbox
,
force_fp32
,
multiclass_nms
)
from
..builder
import
build_loss
from
..losses
import
accuracy
from
..registry
import
HEADS
...
...
mmdet/models/bbox_heads/convfc_bbox_head.py
View file @
d2483e15
import
torch.nn
as
nn
from
.bbox_head
import
BBoxHead
from
..registry
import
HEADS
from
..utils
import
ConvModule
from
.bbox_head
import
BBoxHead
@
HEADS
.
register_module
...
...
mmdet/models/builder.py
View file @
d2483e15
from
torch
import
nn
from
mmdet.utils
import
build_from_cfg
from
.registry
import
(
BACKBONES
,
NECKS
,
ROI_EXTRACTORS
,
SHARED_HEADS
,
HEAD
S
,
LOSSES
,
DETECTOR
S
)
from
.registry
import
(
BACKBONES
,
DETECTORS
,
HEADS
,
LOSSES
,
NECK
S
,
ROI_EXTRACTORS
,
SHARED_HEAD
S
)
def
build
(
cfg
,
registry
,
default_args
=
None
):
...
...
mmdet/models/detectors/base.py
View file @
d2483e15
...
...
@@ -3,10 +3,10 @@ from abc import ABCMeta, abstractmethod
import
mmcv
import
numpy
as
np
import
torch.nn
as
nn
import
pycocotools.mask
as
maskUtils
import
torch.nn
as
nn
from
mmdet.core
import
tensor2imgs
,
get_classes
,
auto_fp16
from
mmdet.core
import
auto_fp16
,
get_classes
,
tensor2imgs
class
BaseDetector
(
nn
.
Module
):
...
...
mmdet/models/detectors/cascade_rcnn.py
View file @
d2483e15
...
...
@@ -3,12 +3,12 @@ from __future__ import division
import
torch
import
torch.nn
as
nn
from
.base
import
BaseDetector
from
.test_mixins
import
RPNTestMixin
from
mmdet.core
import
(
bbox2result
,
bbox2roi
,
build_assigner
,
build_sampler
,
merge_aug_masks
)
from
..
import
builder
from
..registry
import
DETECTORS
from
mmdet.core
import
(
build_assigner
,
bbox2roi
,
bbox2result
,
build_sampler
,
merge_aug_masks
)
from
.base
import
BaseDetector
from
.test_mixins
import
RPNTestMixin
@
DETECTORS
.
register_module
...
...
mmdet/models/detectors/fast_rcnn.py
View file @
d2483e15
from
.two_stage
import
TwoStageDetector
from
..registry
import
DETECTORS
from
.two_stage
import
TwoStageDetector
@
DETECTORS
.
register_module
...
...
mmdet/models/detectors/faster_rcnn.py
View file @
d2483e15
from
.two_stage
import
TwoStageDetector
from
..registry
import
DETECTORS
from
.two_stage
import
TwoStageDetector
@
DETECTORS
.
register_module
...
...
mmdet/models/detectors/fcos.py
View file @
d2483e15
from
.single_stage
import
SingleStageDetector
from
..registry
import
DETECTORS
from
.single_stage
import
SingleStageDetector
@
DETECTORS
.
register_module
...
...
mmdet/models/detectors/grid_rcnn.py
View file @
d2483e15
from
.two_stage
import
TwoStageDetector
from
..registry
import
DETECTORS
import
torch
from
mmdet.core
import
bbox2result
,
bbox2roi
,
build_assigner
,
build_sampler
from
..
import
builder
from
mmdet.core
import
bbox2roi
,
bbox2result
,
build_assigner
,
build_sampler
from
..registry
import
DETECTORS
from
.two_stage
import
TwoStageDetector
@
DETECTORS
.
register_module
...
...
mmdet/models/detectors/htc.py
View file @
d2483e15
import
torch
import
torch.nn.functional
as
F
from
.cascade_rcnn
import
CascadeRCNN
from
mmdet.core
import
(
bbox2result
,
bbox2roi
,
build_assigner
,
build_sampler
,
merge_aug_masks
)
from
..
import
builder
from
..registry
import
DETECTORS
from
mmdet.core
import
(
bbox2roi
,
bbox2result
,
build_assigner
,
build_sampler
,
merge_aug_masks
)
from
.cascade_rcnn
import
CascadeRCNN
@
DETECTORS
.
register_module
...
...
Prev
1
2
3
4
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment