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
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
torch
from
torch.autograd
import
gradcheck
import
os.path
as
osp
import
sys
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
spatial_scale
=
1.0
/
8
...
...
mmdet/ops/roi_pool/gradcheck.py
View file @
d2483e15
import
os.path
as
osp
import
sys
import
torch
from
torch.autograd
import
gradcheck
import
os.path
as
osp
import
sys
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
()
rois
=
torch
.
Tensor
([[
0
,
0
,
0
,
50
,
50
],
[
0
,
10
,
30
,
43
,
55
],
...
...
tools/publish_model.py
View file @
d2483e15
import
argparse
import
subprocess
import
torch
...
...
tools/test.py
View file @
d2483e15
...
...
@@ -7,11 +7,11 @@ import tempfile
import
mmcv
import
torch
import
torch.distributed
as
dist
from
mmcv.runner
import
load_checkpoint
,
get_dist_info
from
mmcv.parallel
import
MMDataParallel
,
MMDistributedDataParallel
from
mmcv.runner
import
get_dist_info
,
load_checkpoint
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.models
import
build_detector
...
...
tools/train.py
View file @
d2483e15
from
__future__
import
division
import
argparse
import
os
import
torch
from
mmcv
import
Config
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.apis
import
(
train_detector
,
init_dist
,
get_root_logger
,
set_random_seed
)
from
mmdet.models
import
build_detector
import
torch
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