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
b6712d4a
Unverified
Commit
b6712d4a
authored
Jul 31, 2019
by
Kai Chen
Committed by
GitHub
Jul 31, 2019
Browse files
Fix the order of import statements in __init__.py (#1094)
* fix the imports in __init__.py * update contribution guide
parent
d2483e15
Changes
28
Show whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
77 additions
and
73 deletions
+77
-73
.travis.yml
.travis.yml
+1
-1
CONTRIBUTING.md
CONTRIBUTING.md
+7
-2
mmdet/apis/__init__.py
mmdet/apis/__init__.py
+2
-2
mmdet/core/anchor/__init__.py
mmdet/core/anchor/__init__.py
+1
-1
mmdet/core/bbox/__init__.py
mmdet/core/bbox/__init__.py
+8
-8
mmdet/core/bbox/assigners/__init__.py
mmdet/core/bbox/assigners/__init__.py
+2
-2
mmdet/core/bbox/samplers/__init__.py
mmdet/core/bbox/samplers/__init__.py
+3
-3
mmdet/core/evaluation/__init__.py
mmdet/core/evaluation/__init__.py
+7
-7
mmdet/core/mask/__init__.py
mmdet/core/mask/__init__.py
+1
-1
mmdet/core/post_processing/__init__.py
mmdet/core/post_processing/__init__.py
+2
-2
mmdet/core/utils/__init__.py
mmdet/core/utils/__init__.py
+2
-2
mmdet/datasets/__init__.py
mmdet/datasets/__init__.py
+8
-8
mmdet/datasets/loader/__init__.py
mmdet/datasets/loader/__init__.py
+1
-1
mmdet/models/__init__.py
mmdet/models/__init__.py
+9
-10
mmdet/models/anchor_heads/__init__.py
mmdet/models/anchor_heads/__init__.py
+3
-3
mmdet/models/backbones/__init__.py
mmdet/models/backbones/__init__.py
+1
-1
mmdet/models/bbox_heads/double_bbox_head.py
mmdet/models/bbox_heads/double_bbox_head.py
+1
-1
mmdet/models/detectors/__init__.py
mmdet/models/detectors/__init__.py
+8
-8
mmdet/models/detectors/double_head_rcnn.py
mmdet/models/detectors/double_head_rcnn.py
+2
-2
mmdet/models/losses/__init__.py
mmdet/models/losses/__init__.py
+8
-8
No files found.
.travis.yml
View file @
b6712d4a
...
@@ -11,5 +11,5 @@ python:
...
@@ -11,5 +11,5 @@ python:
script
:
script
:
-
flake8
-
flake8
-
isort -rc --diff mmdet/ tools/
-
isort -rc
--check-only
--diff mmdet/ tools/
-
yapf -r -d --style .style.yapf mmdet/ tools/
-
yapf -r -d --style .style.yapf mmdet/ tools/
\ No newline at end of file
CONTRIBUTING.md
View file @
b6712d4a
...
@@ -21,8 +21,13 @@ please contact Kai Chen (chenkaidev[at]gmail[dot]com). We will much appreciate y
...
@@ -21,8 +21,13 @@ please contact Kai Chen (chenkaidev[at]gmail[dot]com). We will much appreciate y
### Python
### Python
We adopt
[
PEP8
](
https://www.python.org/dev/peps/pep-0008/
)
as the preferred code style.
We adopt
[
PEP8
](
https://www.python.org/dev/peps/pep-0008/
)
as the preferred code style.
We use
[
flake8
](
http://flake8.pycqa.org/en/latest/
)
as the linter and
[
yapf
](
https://github.com/google/yapf
)
as the formatter.
Please upgrade to the latest yapf (>=0.27.0) and refer to the
[
configuration
](
.style.yapf
)
.
We use the following tools for linting and formatting:
-
[
flake8
](
http://flake8.pycqa.org/en/latest/
)
: linter
-
[
yapf
](
https://github.com/google/yapf
)
: formatter
-
[
isort
](
https://github.com/timothycrosley/isort
)
: sort imports
Style configurations of yapf and isort can be found in
[
.style.yapf
](
.style.yapf
)
and
[
.isort.cfg
](
.isort.cfg
)
.
>Before you create a PR, make sure that your code lints and is formatted by yapf.
>Before you create a PR, make sure that your code lints and is formatted by yapf.
...
...
mmdet/apis/__init__.py
View file @
b6712d4a
from
.env
import
init_dist
,
get_root_logger
,
set_random_seed
from
.env
import
get_root_logger
,
init_dist
,
set_random_seed
from
.inference
import
inference_detector
,
init_detector
,
show_result
from
.train
import
train_detector
from
.train
import
train_detector
from
.inference
import
init_detector
,
inference_detector
,
show_result
__all__
=
[
__all__
=
[
'init_dist'
,
'get_root_logger'
,
'set_random_seed'
,
'train_detector'
,
'init_dist'
,
'get_root_logger'
,
'set_random_seed'
,
'train_detector'
,
...
...
mmdet/core/anchor/__init__.py
View file @
b6712d4a
from
.anchor_generator
import
AnchorGenerator
from
.anchor_generator
import
AnchorGenerator
from
.anchor_target
import
anchor_
target
,
anchor_inside_flags
from
.anchor_target
import
anchor_
inside_flags
,
anchor_target
from
.guided_anchor_target
import
ga_loc_target
,
ga_shape_target
from
.guided_anchor_target
import
ga_loc_target
,
ga_shape_target
__all__
=
[
__all__
=
[
...
...
mmdet/core/bbox/__init__.py
View file @
b6712d4a
from
.assign_sampling
import
assign_and_sample
,
build_assigner
,
build_sampler
from
.assigners
import
AssignResult
,
BaseAssigner
,
MaxIoUAssigner
from
.bbox_target
import
bbox_target
from
.geometry
import
bbox_overlaps
from
.geometry
import
bbox_overlaps
from
.assigners
import
BaseAssigner
,
MaxIoUAssigner
,
AssignResult
from
.samplers
import
(
BaseSampler
,
CombinedSampler
,
from
.samplers
import
(
BaseSampler
,
PseudoSampler
,
RandomSampler
,
InstanceBalancedPosSampler
,
IoUBalancedNegSampler
,
InstanceBalancedPosSampler
,
IoUBalancedNegSampler
,
CombinedSampler
,
SamplingResult
)
PseudoSampler
,
RandomSampler
,
SamplingResult
)
from
.assign_sampling
import
build_assigner
,
build_sampler
,
assign_and_sample
from
.transforms
import
(
bbox2delta
,
bbox2result
,
bbox2roi
,
bbox_flip
,
from
.transforms
import
(
bbox2delta
,
delta2bbox
,
bbox_flip
,
bbox_mapping
,
bbox_mapping
,
bbox_mapping_back
,
delta2bbox
,
bbox_mapping_back
,
bbox2roi
,
roi2bbox
,
bbox2result
,
distance2bbox
,
roi2bbox
)
distance2bbox
)
from
.bbox_target
import
bbox_target
__all__
=
[
__all__
=
[
'bbox_overlaps'
,
'BaseAssigner'
,
'MaxIoUAssigner'
,
'AssignResult'
,
'bbox_overlaps'
,
'BaseAssigner'
,
'MaxIoUAssigner'
,
'AssignResult'
,
...
...
mmdet/core/bbox/assigners/__init__.py
View file @
b6712d4a
from
.base_assigner
import
BaseAssigner
from
.max_iou_assigner
import
MaxIoUAssigner
from
.approx_max_iou_assigner
import
ApproxMaxIoUAssigner
from
.approx_max_iou_assigner
import
ApproxMaxIoUAssigner
from
.assign_result
import
AssignResult
from
.assign_result
import
AssignResult
from
.base_assigner
import
BaseAssigner
from
.max_iou_assigner
import
MaxIoUAssigner
__all__
=
[
__all__
=
[
'BaseAssigner'
,
'MaxIoUAssigner'
,
'ApproxMaxIoUAssigner'
,
'AssignResult'
'BaseAssigner'
,
'MaxIoUAssigner'
,
'ApproxMaxIoUAssigner'
,
'AssignResult'
...
...
mmdet/core/bbox/samplers/__init__.py
View file @
b6712d4a
from
.base_sampler
import
BaseSampler
from
.base_sampler
import
BaseSampler
from
.pseudo_sampler
import
PseudoSampler
from
.combined_sampler
import
CombinedSampler
from
.random_sampler
import
RandomSampler
from
.instance_balanced_pos_sampler
import
InstanceBalancedPosSampler
from
.instance_balanced_pos_sampler
import
InstanceBalancedPosSampler
from
.iou_balanced_neg_sampler
import
IoUBalancedNegSampler
from
.iou_balanced_neg_sampler
import
IoUBalancedNegSampler
from
.combined_sampler
import
CombinedSampler
from
.ohem_sampler
import
OHEMSampler
from
.ohem_sampler
import
OHEMSampler
from
.pseudo_sampler
import
PseudoSampler
from
.random_sampler
import
RandomSampler
from
.sampling_result
import
SamplingResult
from
.sampling_result
import
SamplingResult
__all__
=
[
__all__
=
[
...
...
mmdet/core/evaluation/__init__.py
View file @
b6712d4a
from
.class_names
import
(
v
oc_classes
,
imagenet_d
et_classes
,
from
.class_names
import
(
c
oc
o
_classes
,
dataset_aliases
,
g
et_classes
,
imagenet_
vid
_classes
,
coco_classes
,
dataset_alia
ses
,
imagenet_
det
_classes
,
imagenet_vid_clas
ses
,
get
_classes
)
voc
_classes
)
from
.coco_utils
import
coco_eval
,
fast_eval_recall
,
results2json
from
.coco_utils
import
coco_eval
,
fast_eval_recall
,
results2json
from
.eval_hooks
import
(
DistEvalHook
,
DistEvalmAPHook
,
CocoDistEvalRecallHook
,
from
.eval_hooks
import
(
Coco
DistEvalmAPHook
,
CocoDistEvalRecallHook
,
Coco
DistEvalmAPHook
)
DistEvalHook
,
DistEvalmAPHook
)
from
.mean_ap
import
average_precision
,
eval_map
,
print_map_summary
from
.mean_ap
import
average_precision
,
eval_map
,
print_map_summary
from
.recall
import
(
eval_recalls
,
p
rint_recall_summary
,
plot_num_recall
,
from
.recall
import
(
eval_recalls
,
p
lot_iou_recall
,
plot_num_recall
,
p
lot_iou_recall
)
p
rint_recall_summary
)
__all__
=
[
__all__
=
[
'voc_classes'
,
'imagenet_det_classes'
,
'imagenet_vid_classes'
,
'voc_classes'
,
'imagenet_det_classes'
,
'imagenet_vid_classes'
,
...
...
mmdet/core/mask/__init__.py
View file @
b6712d4a
from
.utils
import
split_combined_polys
from
.mask_target
import
mask_target
from
.mask_target
import
mask_target
from
.utils
import
split_combined_polys
__all__
=
[
'split_combined_polys'
,
'mask_target'
]
__all__
=
[
'split_combined_polys'
,
'mask_target'
]
mmdet/core/post_processing/__init__.py
View file @
b6712d4a
from
.bbox_nms
import
multiclass_nms
from
.bbox_nms
import
multiclass_nms
from
.merge_augs
import
(
merge_aug_
proposal
s
,
merge_aug_
bboxe
s
,
from
.merge_augs
import
(
merge_aug_
bboxe
s
,
merge_aug_
mask
s
,
merge_aug_
score
s
,
merge_aug_
mask
s
)
merge_aug_
proposal
s
,
merge_aug_
score
s
)
__all__
=
[
__all__
=
[
'multiclass_nms'
,
'merge_aug_proposals'
,
'merge_aug_bboxes'
,
'multiclass_nms'
,
'merge_aug_proposals'
,
'merge_aug_bboxes'
,
...
...
mmdet/core/utils/__init__.py
View file @
b6712d4a
from
.dist_utils
import
allreduce_grads
,
DistOptimizerHook
from
.dist_utils
import
DistOptimizerHook
,
allreduce_grads
from
.misc
import
tensor2imgs
,
unmap
,
multi_apply
from
.misc
import
multi_apply
,
tensor2imgs
,
unmap
__all__
=
[
__all__
=
[
'allreduce_grads'
,
'DistOptimizerHook'
,
'tensor2imgs'
,
'unmap'
,
'allreduce_grads'
,
'DistOptimizerHook'
,
'tensor2imgs'
,
'unmap'
,
...
...
mmdet/datasets/__init__.py
View file @
b6712d4a
from
.custom
import
CustomDataset
from
.builder
import
build_dataset
from
.xml_style
import
XMLDataset
from
.coco
import
CocoDataset
from
.cityscapes
import
CityscapesDataset
from
.cityscapes
import
CityscapesDataset
from
.voc
import
VOCDataset
from
.coco
import
CocoDataset
from
.wider_face
import
WIDERFaceDataset
from
.custom
import
CustomDataset
from
.loader
import
GroupSampler
,
DistributedGroupSampler
,
build_dataloader
from
.utils
import
to_tensor
,
random_scale
,
show_ann
from
.dataset_wrappers
import
ConcatDataset
,
RepeatDataset
from
.dataset_wrappers
import
ConcatDataset
,
RepeatDataset
from
.extra_aug
import
ExtraAugmentation
from
.extra_aug
import
ExtraAugmentation
from
.loader
import
DistributedGroupSampler
,
GroupSampler
,
build_dataloader
from
.registry
import
DATASETS
from
.registry
import
DATASETS
from
.builder
import
build_dataset
from
.utils
import
random_scale
,
show_ann
,
to_tensor
from
.voc
import
VOCDataset
from
.wider_face
import
WIDERFaceDataset
from
.xml_style
import
XMLDataset
__all__
=
[
__all__
=
[
'CustomDataset'
,
'XMLDataset'
,
'CocoDataset'
,
'VOCDataset'
,
'CustomDataset'
,
'XMLDataset'
,
'CocoDataset'
,
'VOCDataset'
,
...
...
mmdet/datasets/loader/__init__.py
View file @
b6712d4a
from
.build_loader
import
build_dataloader
from
.build_loader
import
build_dataloader
from
.sampler
import
GroupSampler
,
Distributed
GroupSampler
from
.sampler
import
Distributed
GroupSampler
,
GroupSampler
__all__
=
[
'GroupSampler'
,
'DistributedGroupSampler'
,
'build_dataloader'
]
__all__
=
[
'GroupSampler'
,
'DistributedGroupSampler'
,
'build_dataloader'
]
mmdet/models/__init__.py
View file @
b6712d4a
from
.anchor_heads
import
*
# noqa: F401,F403
from
.backbones
import
*
# noqa: F401,F403
from
.backbones
import
*
# noqa: F401,F403
from
.bbox_heads
import
*
# noqa: F401,F403
from
.builder
import
(
build_backbone
,
build_detector
,
build_head
,
build_loss
,
build_neck
,
build_roi_extractor
,
build_shared_head
)
from
.detectors
import
*
# noqa: F401,F403
from
.losses
import
*
# noqa: F401,F403
from
.mask_heads
import
*
# noqa: F401,F403
from
.necks
import
*
# noqa: F401,F403
from
.necks
import
*
# noqa: F401,F403
from
.registry
import
(
BACKBONES
,
DETECTORS
,
HEADS
,
LOSSES
,
NECKS
,
ROI_EXTRACTORS
,
SHARED_HEADS
)
from
.roi_extractors
import
*
# noqa: F401,F403
from
.roi_extractors
import
*
# noqa: F401,F403
from
.anchor_heads
import
*
# noqa: F401,F403
from
.shared_heads
import
*
# noqa: F401,F403
from
.shared_heads
import
*
# noqa: F401,F403
from
.bbox_heads
import
*
# noqa: F401,F403
from
.mask_heads
import
*
# noqa: F401,F403
from
.losses
import
*
# noqa: F401,F403
from
.detectors
import
*
# noqa: F401,F403
from
.registry
import
(
BACKBONES
,
NECKS
,
ROI_EXTRACTORS
,
SHARED_HEADS
,
HEADS
,
LOSSES
,
DETECTORS
)
from
.builder
import
(
build_backbone
,
build_neck
,
build_roi_extractor
,
build_shared_head
,
build_head
,
build_loss
,
build_detector
)
__all__
=
[
__all__
=
[
'BACKBONES'
,
'NECKS'
,
'ROI_EXTRACTORS'
,
'SHARED_HEADS'
,
'HEADS'
,
'LOSSES'
,
'BACKBONES'
,
'NECKS'
,
'ROI_EXTRACTORS'
,
'SHARED_HEADS'
,
'HEADS'
,
'LOSSES'
,
...
...
mmdet/models/anchor_heads/__init__.py
View file @
b6712d4a
from
.anchor_head
import
AnchorHead
from
.anchor_head
import
AnchorHead
from
.guided_anchor_head
import
GuidedAnchorHead
,
FeatureAdaption
from
.fcos_head
import
FCOSHead
from
.fcos_head
import
FCOSHead
from
.
rpn
_head
import
RPN
Head
from
.
ga_retina
_head
import
GARetina
Head
from
.ga_rpn_head
import
GARPNHead
from
.ga_rpn_head
import
GARPNHead
from
.guided_anchor_head
import
FeatureAdaption
,
GuidedAnchorHead
from
.retina_head
import
RetinaHead
from
.retina_head
import
RetinaHead
from
.
ga_retina
_head
import
GARetina
Head
from
.
rpn
_head
import
RPN
Head
from
.ssd_head
import
SSDHead
from
.ssd_head
import
SSDHead
__all__
=
[
__all__
=
[
...
...
mmdet/models/backbones/__init__.py
View file @
b6712d4a
from
.hrnet
import
HRNet
from
.resnet
import
ResNet
,
make_res_layer
from
.resnet
import
ResNet
,
make_res_layer
from
.resnext
import
ResNeXt
from
.resnext
import
ResNeXt
from
.ssd_vgg
import
SSDVGG
from
.ssd_vgg
import
SSDVGG
from
.hrnet
import
HRNet
__all__
=
[
'ResNet'
,
'make_res_layer'
,
'ResNeXt'
,
'SSDVGG'
,
'HRNet'
]
__all__
=
[
'ResNet'
,
'make_res_layer'
,
'ResNeXt'
,
'SSDVGG'
,
'HRNet'
]
mmdet/models/bbox_heads/double_bbox_head.py
View file @
b6712d4a
import
torch.nn
as
nn
import
torch.nn
as
nn
from
mmcv.cnn.weight_init
import
normal_init
,
xavier_init
from
mmcv.cnn.weight_init
import
normal_init
,
xavier_init
from
.bbox_head
import
BBoxHead
from
..backbones.resnet
import
Bottleneck
from
..backbones.resnet
import
Bottleneck
from
..registry
import
HEADS
from
..registry
import
HEADS
from
..utils
import
ConvModule
from
..utils
import
ConvModule
from
.bbox_head
import
BBoxHead
class
BasicResBlock
(
nn
.
Module
):
class
BasicResBlock
(
nn
.
Module
):
...
...
mmdet/models/detectors/__init__.py
View file @
b6712d4a
from
.base
import
BaseDetector
from
.base
import
BaseDetector
from
.single_stage
import
SingleStageDetector
from
.two_stage
import
TwoStageDetector
from
.rpn
import
RPN
from
.fast_rcnn
import
FastRCNN
from
.faster_rcnn
import
FasterRCNN
from
.mask_rcnn
import
MaskRCNN
from
.cascade_rcnn
import
CascadeRCNN
from
.cascade_rcnn
import
CascadeRCNN
from
.double_head_rcnn
import
DoubleHeadRCNN
from
.double_head_rcnn
import
DoubleHeadRCNN
from
.
htc
import
HybridTaskCascade
from
.
fast_rcnn
import
FastRCNN
from
.
retinanet
import
RetinaNet
from
.
faster_rcnn
import
FasterRCNN
from
.fcos
import
FCOS
from
.fcos
import
FCOS
from
.grid_rcnn
import
GridRCNN
from
.grid_rcnn
import
GridRCNN
from
.htc
import
HybridTaskCascade
from
.mask_rcnn
import
MaskRCNN
from
.mask_scoring_rcnn
import
MaskScoringRCNN
from
.mask_scoring_rcnn
import
MaskScoringRCNN
from
.retinanet
import
RetinaNet
from
.rpn
import
RPN
from
.single_stage
import
SingleStageDetector
from
.two_stage
import
TwoStageDetector
__all__
=
[
__all__
=
[
'BaseDetector'
,
'SingleStageDetector'
,
'TwoStageDetector'
,
'RPN'
,
'BaseDetector'
,
'SingleStageDetector'
,
'TwoStageDetector'
,
'RPN'
,
...
...
mmdet/models/detectors/double_head_rcnn.py
View file @
b6712d4a
import
torch
import
torch
from
.two_stage
import
TwoStageDetector
from
..registry
import
DETECTORS
from
mmdet.core
import
bbox2roi
,
build_assigner
,
build_sampler
from
mmdet.core
import
bbox2roi
,
build_assigner
,
build_sampler
from
..registry
import
DETECTORS
from
.two_stage
import
TwoStageDetector
@
DETECTORS
.
register_module
@
DETECTORS
.
register_module
...
...
mmdet/models/losses/__init__.py
View file @
b6712d4a
from
.accuracy
import
a
ccuracy
,
A
ccuracy
from
.accuracy
import
A
ccuracy
,
a
ccuracy
from
.
cross_entropy_loss
import
(
cross_entropy
,
binary_cross_entropy
,
from
.
balanced_l1_loss
import
BalancedL1Loss
,
balanced_l1_loss
mask_cross_entropy
,
CrossEntropyLoss
)
from
.cross_entropy_loss
import
(
CrossEntropyLoss
,
binary_cross_entropy
,
from
.focal_loss
import
sigmoid_focal_loss
,
FocalLoss
cross_entropy
,
mask_cross_entropy
)
from
.
smooth_l1
_loss
import
smooth_l1_loss
,
SmoothL1L
oss
from
.
focal
_loss
import
FocalLoss
,
sigmoid_focal_l
oss
from
.ghm_loss
import
GHMC
,
GHMR
from
.ghm_loss
import
GHMC
,
GHMR
from
.
balanced_l1
_loss
import
balanced_l1_loss
,
BalancedL1L
oss
from
.
iou
_loss
import
BoundedIoULoss
,
IoULoss
,
bounded_iou_loss
,
iou_l
oss
from
.mse_loss
import
mse_loss
,
MSEL
oss
from
.mse_loss
import
MSELoss
,
mse_l
oss
from
.
iou
_loss
import
iou_loss
,
bounded_iou_loss
,
IoULoss
,
BoundedIoUL
oss
from
.
smooth_l1
_loss
import
SmoothL1Loss
,
smooth_l1_l
oss
from
.utils
import
reduce_loss
,
weight_reduce_loss
,
weighted_loss
from
.utils
import
reduce_loss
,
weight_reduce_loss
,
weighted_loss
__all__
=
[
__all__
=
[
...
...
Prev
1
2
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