Commit 6e238d52 authored by Kai Chen's avatar Kai Chen
Browse files

minor fix

parent 82356fd9
...@@ -95,7 +95,7 @@ class MaskTransform(object): ...@@ -95,7 +95,7 @@ class MaskTransform(object):
for mask in masks for mask in masks
] ]
if flip: if flip:
masks = [mmcv.imflip(mask) for mask in masks] masks = [mask[:, ::-1] for mask in masks]
padded_masks = [ padded_masks = [
mmcv.impad(mask, pad_shape[:2], pad_val=0) for mask in masks mmcv.impad(mask, pad_shape[:2], pad_val=0) for mask in masks
] ]
......
...@@ -8,12 +8,12 @@ from collections import OrderedDict ...@@ -8,12 +8,12 @@ from collections import OrderedDict
import numpy as np import numpy as np
import torch import torch
from mmcv import Config 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 import datasets, __version__
from mmdet.core import (init_dist, DistOptimizerHook, DistSamplerSeedHook, from mmdet.core import (init_dist, DistOptimizerHook, MMDataParallel,
MMDataParallel, MMDistributedDataParallel, MMDistributedDataParallel, CocoDistEvalRecallHook,
CocoDistEvalRecallHook, CocoDistEvalmAPHook) CocoDistEvalmAPHook)
from mmdet.datasets import build_dataloader from mmdet.datasets import build_dataloader
from mmdet.models import build_detector, RPN from mmdet.models import build_detector, RPN
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment