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

minor fix

parent 82356fd9
......@@ -95,7 +95,7 @@ class MaskTransform(object):
for mask in masks
]
if flip:
masks = [mmcv.imflip(mask) for mask in masks]
masks = [mask[:, ::-1] for mask in masks]
padded_masks = [
mmcv.impad(mask, pad_shape[:2], pad_val=0) for mask in masks
]
......
......@@ -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
......
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