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
5 changed files
with
15 additions
and
12 deletions
+15
-12
mmdet/ops/roi_align/gradcheck.py
mmdet/ops/roi_align/gradcheck.py
+4
-3
mmdet/ops/roi_pool/gradcheck.py
mmdet/ops/roi_pool/gradcheck.py
+4
-3
tools/publish_model.py
tools/publish_model.py
+1
-0
tools/test.py
tools/test.py
+2
-2
tools/train.py
tools/train.py
+4
-4
No files found.
mmdet/ops/roi_align/gradcheck.py
View file @
d2483e15
import
os.path
as
osp
import
sys
import
numpy
as
np
import
numpy
as
np
import
torch
import
torch
from
torch.autograd
import
gradcheck
from
torch.autograd
import
gradcheck
import
os.path
as
osp
import
sys
sys
.
path
.
append
(
osp
.
abspath
(
osp
.
join
(
__file__
,
'../../'
)))
sys
.
path
.
append
(
osp
.
abspath
(
osp
.
join
(
__file__
,
'../../'
)))
from
roi_align
import
RoIAlign
# noqa: E402
from
roi_align
import
RoIAlign
# noqa: E402
, isort:skip
feat_size
=
15
feat_size
=
15
spatial_scale
=
1.0
/
8
spatial_scale
=
1.0
/
8
...
...
mmdet/ops/roi_pool/gradcheck.py
View file @
d2483e15
import
os.path
as
osp
import
sys
import
torch
import
torch
from
torch.autograd
import
gradcheck
from
torch.autograd
import
gradcheck
import
os.path
as
osp
import
sys
sys
.
path
.
append
(
osp
.
abspath
(
osp
.
join
(
__file__
,
'../../'
)))
sys
.
path
.
append
(
osp
.
abspath
(
osp
.
join
(
__file__
,
'../../'
)))
from
roi_pool
import
RoIPool
# noqa: E402
from
roi_pool
import
RoIPool
# noqa: E402
, isort:skip
feat
=
torch
.
randn
(
4
,
16
,
15
,
15
,
requires_grad
=
True
).
cuda
()
feat
=
torch
.
randn
(
4
,
16
,
15
,
15
,
requires_grad
=
True
).
cuda
()
rois
=
torch
.
Tensor
([[
0
,
0
,
0
,
50
,
50
],
[
0
,
10
,
30
,
43
,
55
],
rois
=
torch
.
Tensor
([[
0
,
0
,
0
,
50
,
50
],
[
0
,
10
,
30
,
43
,
55
],
...
...
tools/publish_model.py
View file @
d2483e15
import
argparse
import
argparse
import
subprocess
import
subprocess
import
torch
import
torch
...
...
tools/test.py
View file @
d2483e15
...
@@ -7,11 +7,11 @@ import tempfile
...
@@ -7,11 +7,11 @@ import tempfile
import
mmcv
import
mmcv
import
torch
import
torch
import
torch.distributed
as
dist
import
torch.distributed
as
dist
from
mmcv.runner
import
load_checkpoint
,
get_dist_info
from
mmcv.parallel
import
MMDataParallel
,
MMDistributedDataParallel
from
mmcv.parallel
import
MMDataParallel
,
MMDistributedDataParallel
from
mmcv.runner
import
get_dist_info
,
load_checkpoint
from
mmdet.apis
import
init_dist
from
mmdet.apis
import
init_dist
from
mmdet.core
import
results2json
,
coco_eval
,
wrap_fp16_model
from
mmdet.core
import
coco_eval
,
results2json
,
wrap_fp16_model
from
mmdet.datasets
import
build_dataloader
,
build_dataset
from
mmdet.datasets
import
build_dataloader
,
build_dataset
from
mmdet.models
import
build_detector
from
mmdet.models
import
build_detector
...
...
tools/train.py
View file @
d2483e15
from
__future__
import
division
from
__future__
import
division
import
argparse
import
argparse
import
os
import
os
import
torch
from
mmcv
import
Config
from
mmcv
import
Config
from
mmdet
import
__version__
from
mmdet
import
__version__
from
mmdet.apis
import
(
get_root_logger
,
init_dist
,
set_random_seed
,
train_detector
)
from
mmdet.datasets
import
build_dataset
from
mmdet.datasets
import
build_dataset
from
mmdet.apis
import
(
train_detector
,
init_dist
,
get_root_logger
,
set_random_seed
)
from
mmdet.models
import
build_detector
from
mmdet.models
import
build_detector
import
torch
def
parse_args
():
def
parse_args
():
...
...
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