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
Show whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
29 additions
and
29 deletions
+29
-29
mmdet/models/detectors/mask_rcnn.py
mmdet/models/detectors/mask_rcnn.py
+1
-1
mmdet/models/detectors/mask_scoring_rcnn.py
mmdet/models/detectors/mask_scoring_rcnn.py
+1
-1
mmdet/models/detectors/retinanet.py
mmdet/models/detectors/retinanet.py
+1
-1
mmdet/models/detectors/rpn.py
mmdet/models/detectors/rpn.py
+3
-3
mmdet/models/detectors/single_stage.py
mmdet/models/detectors/single_stage.py
+2
-2
mmdet/models/detectors/test_mixins.py
mmdet/models/detectors/test_mixins.py
+2
-2
mmdet/models/detectors/two_stage.py
mmdet/models/detectors/two_stage.py
+3
-3
mmdet/models/losses/balanced_l1_loss.py
mmdet/models/losses/balanced_l1_loss.py
+1
-1
mmdet/models/losses/cross_entropy_loss.py
mmdet/models/losses/cross_entropy_loss.py
+1
-1
mmdet/models/losses/focal_loss.py
mmdet/models/losses/focal_loss.py
+1
-1
mmdet/models/losses/iou_loss.py
mmdet/models/losses/iou_loss.py
+1
-1
mmdet/models/losses/mse_loss.py
mmdet/models/losses/mse_loss.py
+1
-1
mmdet/models/losses/smooth_l1_loss.py
mmdet/models/losses/smooth_l1_loss.py
+1
-1
mmdet/models/mask_heads/fcn_mask_head.py
mmdet/models/mask_heads/fcn_mask_head.py
+1
-1
mmdet/models/mask_heads/htc_mask_head.py
mmdet/models/mask_heads/htc_mask_head.py
+1
-1
mmdet/models/mask_heads/maskiou_head.py
mmdet/models/mask_heads/maskiou_head.py
+1
-1
mmdet/models/necks/hrfpn.py
mmdet/models/necks/hrfpn.py
+2
-2
mmdet/models/plugins/generalized_attention.py
mmdet/models/plugins/generalized_attention.py
+3
-3
mmdet/models/utils/conv_module.py
mmdet/models/utils/conv_module.py
+1
-1
mmdet/ops/nms/nms_wrapper.py
mmdet/ops/nms/nms_wrapper.py
+1
-1
No files found.
mmdet/models/detectors/mask_rcnn.py
View file @
d2483e15
from
.two_stage
import
TwoStageDetector
from
..registry
import
DETECTORS
from
..registry
import
DETECTORS
from
.two_stage
import
TwoStageDetector
@
DETECTORS
.
register_module
@
DETECTORS
.
register_module
...
...
mmdet/models/detectors/mask_scoring_rcnn.py
View file @
d2483e15
import
torch
import
torch
from
mmdet.core
import
bbox2roi
,
build_assigner
,
build_sampler
from
mmdet.core
import
bbox2roi
,
build_assigner
,
build_sampler
from
.two_stage
import
TwoStageDetector
from
..
import
builder
from
..
import
builder
from
..registry
import
DETECTORS
from
..registry
import
DETECTORS
from
.two_stage
import
TwoStageDetector
@
DETECTORS
.
register_module
@
DETECTORS
.
register_module
...
...
mmdet/models/detectors/retinanet.py
View file @
d2483e15
from
.single_stage
import
SingleStageDetector
from
..registry
import
DETECTORS
from
..registry
import
DETECTORS
from
.single_stage
import
SingleStageDetector
@
DETECTORS
.
register_module
@
DETECTORS
.
register_module
...
...
mmdet/models/detectors/rpn.py
View file @
d2483e15
import
mmcv
import
mmcv
from
mmdet.core
import
tensor2imgs
,
bbox_mapping
from
mmdet.core
import
bbox_mapping
,
tensor2imgs
from
.base
import
BaseDetector
from
.test_mixins
import
RPNTestMixin
from
..
import
builder
from
..
import
builder
from
..registry
import
DETECTORS
from
..registry
import
DETECTORS
from
.base
import
BaseDetector
from
.test_mixins
import
RPNTestMixin
@
DETECTORS
.
register_module
@
DETECTORS
.
register_module
...
...
mmdet/models/detectors/single_stage.py
View file @
d2483e15
import
torch.nn
as
nn
import
torch.nn
as
nn
from
.base
import
BaseDetector
from
mmdet.core
import
bbox2result
from
..
import
builder
from
..
import
builder
from
..registry
import
DETECTORS
from
..registry
import
DETECTORS
from
mmdet.core
import
bbox2result
from
.base
import
BaseDetector
@
DETECTORS
.
register_module
@
DETECTORS
.
register_module
...
...
mmdet/models/detectors/test_mixins.py
View file @
d2483e15
from
mmdet.core
import
(
bbox2roi
,
bbox_mapping
,
merge_aug_
proposal
s
,
from
mmdet.core
import
(
bbox2roi
,
bbox_mapping
,
merge_aug_
bboxe
s
,
merge_aug_
bboxe
s
,
merge_aug_
mask
s
,
multiclass_nms
)
merge_aug_
mask
s
,
merge_aug_
proposal
s
,
multiclass_nms
)
class
RPNTestMixin
(
object
):
class
RPNTestMixin
(
object
):
...
...
mmdet/models/detectors/two_stage.py
View file @
d2483e15
import
torch
import
torch
import
torch.nn
as
nn
import
torch.nn
as
nn
from
.base
import
BaseDetector
from
mmdet.core
import
bbox2result
,
bbox2roi
,
build_assigner
,
build_sampler
from
.test_mixins
import
RPNTestMixin
,
BBoxTestMixin
,
MaskTestMixin
from
..
import
builder
from
..
import
builder
from
..registry
import
DETECTORS
from
..registry
import
DETECTORS
from
mmdet.core
import
bbox2roi
,
bbox2result
,
build_assigner
,
build_sampler
from
.base
import
BaseDetector
from
.test_mixins
import
BBoxTestMixin
,
MaskTestMixin
,
RPNTestMixin
@
DETECTORS
.
register_module
@
DETECTORS
.
register_module
...
...
mmdet/models/losses/balanced_l1_loss.py
View file @
d2483e15
...
@@ -2,8 +2,8 @@ import numpy as np
...
@@ -2,8 +2,8 @@ import numpy as np
import
torch
import
torch
import
torch.nn
as
nn
import
torch.nn
as
nn
from
.utils
import
weighted_loss
from
..registry
import
LOSSES
from
..registry
import
LOSSES
from
.utils
import
weighted_loss
@
weighted_loss
@
weighted_loss
...
...
mmdet/models/losses/cross_entropy_loss.py
View file @
d2483e15
...
@@ -2,8 +2,8 @@ import torch
...
@@ -2,8 +2,8 @@ import torch
import
torch.nn
as
nn
import
torch.nn
as
nn
import
torch.nn.functional
as
F
import
torch.nn.functional
as
F
from
.utils
import
weight_reduce_loss
from
..registry
import
LOSSES
from
..registry
import
LOSSES
from
.utils
import
weight_reduce_loss
def
cross_entropy
(
pred
,
label
,
weight
=
None
,
reduction
=
'mean'
,
avg_factor
=
None
):
def
cross_entropy
(
pred
,
label
,
weight
=
None
,
reduction
=
'mean'
,
avg_factor
=
None
):
...
...
mmdet/models/losses/focal_loss.py
View file @
d2483e15
...
@@ -2,8 +2,8 @@ import torch.nn as nn
...
@@ -2,8 +2,8 @@ import torch.nn as nn
import
torch.nn.functional
as
F
import
torch.nn.functional
as
F
from
mmdet.ops
import
sigmoid_focal_loss
as
_sigmoid_focal_loss
from
mmdet.ops
import
sigmoid_focal_loss
as
_sigmoid_focal_loss
from
.utils
import
weight_reduce_loss
from
..registry
import
LOSSES
from
..registry
import
LOSSES
from
.utils
import
weight_reduce_loss
# This method is only for debugging
# This method is only for debugging
...
...
mmdet/models/losses/iou_loss.py
View file @
d2483e15
...
@@ -2,8 +2,8 @@ import torch
...
@@ -2,8 +2,8 @@ import torch
import
torch.nn
as
nn
import
torch.nn
as
nn
from
mmdet.core
import
bbox_overlaps
from
mmdet.core
import
bbox_overlaps
from
.utils
import
weighted_loss
from
..registry
import
LOSSES
from
..registry
import
LOSSES
from
.utils
import
weighted_loss
@
weighted_loss
@
weighted_loss
...
...
mmdet/models/losses/mse_loss.py
View file @
d2483e15
import
torch.nn
as
nn
import
torch.nn
as
nn
import
torch.nn.functional
as
F
import
torch.nn.functional
as
F
from
.utils
import
weighted_loss
from
..registry
import
LOSSES
from
..registry
import
LOSSES
from
.utils
import
weighted_loss
mse_loss
=
weighted_loss
(
F
.
mse_loss
)
mse_loss
=
weighted_loss
(
F
.
mse_loss
)
...
...
mmdet/models/losses/smooth_l1_loss.py
View file @
d2483e15
import
torch
import
torch
import
torch.nn
as
nn
import
torch.nn
as
nn
from
.utils
import
weighted_loss
from
..registry
import
LOSSES
from
..registry
import
LOSSES
from
.utils
import
weighted_loss
@
weighted_loss
@
weighted_loss
...
...
mmdet/models/mask_heads/fcn_mask_head.py
View file @
d2483e15
...
@@ -4,10 +4,10 @@ import pycocotools.mask as mask_util
...
@@ -4,10 +4,10 @@ import pycocotools.mask as mask_util
import
torch
import
torch
import
torch.nn
as
nn
import
torch.nn
as
nn
from
mmdet.core
import
auto_fp16
,
force_fp32
,
mask_target
from
..builder
import
build_loss
from
..builder
import
build_loss
from
..registry
import
HEADS
from
..registry
import
HEADS
from
..utils
import
ConvModule
from
..utils
import
ConvModule
from
mmdet.core
import
mask_target
,
force_fp32
,
auto_fp16
@
HEADS
.
register_module
@
HEADS
.
register_module
...
...
mmdet/models/mask_heads/htc_mask_head.py
View file @
d2483e15
from
.fcn_mask_head
import
FCNMaskHead
from
..registry
import
HEADS
from
..registry
import
HEADS
from
..utils
import
ConvModule
from
..utils
import
ConvModule
from
.fcn_mask_head
import
FCNMaskHead
@
HEADS
.
register_module
@
HEADS
.
register_module
...
...
mmdet/models/mask_heads/maskiou_head.py
View file @
d2483e15
...
@@ -2,8 +2,8 @@ import numpy as np
...
@@ -2,8 +2,8 @@ import numpy as np
import
torch
import
torch
import
torch.nn
as
nn
import
torch.nn
as
nn
from
mmcv.cnn
import
kaiming_init
,
normal_init
from
mmcv.cnn
import
kaiming_init
,
normal_init
from
mmdet.core
import
force_fp32
from
mmdet.core
import
force_fp32
from
..builder
import
build_loss
from
..builder
import
build_loss
from
..registry
import
HEADS
from
..registry
import
HEADS
...
...
mmdet/models/necks/hrfpn.py
View file @
d2483e15
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
from
torch.utils.checkpoint
import
checkpoint
from
mmcv.cnn.weight_init
import
caffe2_xavier_init
from
mmcv.cnn.weight_init
import
caffe2_xavier_init
from
torch.utils.checkpoint
import
checkpoint
from
..utils
import
ConvModule
from
..registry
import
NECKS
from
..registry
import
NECKS
from
..utils
import
ConvModule
@
NECKS
.
register_module
@
NECKS
.
register_module
...
...
mmdet/models/plugins/generalized_attention.py
View file @
d2483e15
import
math
import
numpy
as
np
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
import
math
import
numpy
as
np
from
mmcv.cnn
import
kaiming_init
from
mmcv.cnn
import
kaiming_init
...
...
mmdet/models/utils/conv_module.py
View file @
d2483e15
import
warnings
import
warnings
import
torch.nn
as
nn
import
torch.nn
as
nn
from
mmcv.cnn
import
kaiming_init
,
constant
_init
from
mmcv.cnn
import
constant_init
,
kaiming
_init
from
.conv_ws
import
ConvWS2d
from
.conv_ws
import
ConvWS2d
from
.norm
import
build_norm_layer
from
.norm
import
build_norm_layer
...
...
mmdet/ops/nms/nms_wrapper.py
View file @
d2483e15
import
numpy
as
np
import
numpy
as
np
import
torch
import
torch
from
.
import
nms_cu
da
,
nms_c
p
u
from
.
import
nms_c
p
u
,
nms_cu
da
from
.soft_nms_cpu
import
soft_nms_cpu
from
.soft_nms_cpu
import
soft_nms_cpu
...
...
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