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
6e238d52
"vscode:/vscode.git/clone" did not exist on "2d60a5ee75a8506bf7ee97090afc3f0d1baf88d4"
Commit
6e238d52
authored
Oct 08, 2018
by
Kai Chen
Browse files
minor fix
parent
82356fd9
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
5 deletions
+5
-5
mmdet/datasets/transforms.py
mmdet/datasets/transforms.py
+1
-1
tools/train.py
tools/train.py
+4
-4
No files found.
mmdet/datasets/transforms.py
View file @
6e238d52
...
...
@@ -95,7 +95,7 @@ class MaskTransform(object):
for
mask
in
masks
]
if
flip
:
masks
=
[
m
mcv
.
imflip
(
mask
)
for
mask
in
masks
]
masks
=
[
m
ask
[:,
::
-
1
]
for
mask
in
masks
]
padded_masks
=
[
mmcv
.
impad
(
mask
,
pad_shape
[:
2
],
pad_val
=
0
)
for
mask
in
masks
]
...
...
tools/train.py
View file @
6e238d52
...
...
@@ -8,12 +8,12 @@ from collections import OrderedDict
import
numpy
as
np
import
torch
from
mmcv
import
Config
from
mmcv.runner
import
Runner
,
obj_from_dict
from
mmcv.runner
import
Runner
,
obj_from_dict
,
DistSamplerSeedHook
from
mmdet
import
datasets
,
__version__
from
mmdet.core
import
(
init_dist
,
DistOptimizerHook
,
DistSamplerSeedHook
,
MMDataParallel
,
MMDistributedDataParallel
,
CocoDistEvalRecallHook
,
CocoDistEvalmAPHook
)
from
mmdet.core
import
(
init_dist
,
DistOptimizerHook
,
MMDataParallel
,
MMDistributedDataParallel
,
CocoDistEvalRecallHook
,
CocoDistEvalmAPHook
)
from
mmdet.datasets
import
build_dataloader
from
mmdet.models
import
build_detector
,
RPN
...
...
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