Commit 6b25743a authored by wangg12's avatar wangg12
Browse files

fix flake8

parent 7cbdbc78
...@@ -5,6 +5,7 @@ from .utils import to_tensor, random_scale, show_ann, get_dataset ...@@ -5,6 +5,7 @@ from .utils import to_tensor, random_scale, show_ann, get_dataset
from .concat_dataset import ConcatDataset from .concat_dataset import ConcatDataset
__all__ = [ __all__ = [
'CustomDataset', 'CocoDataset', 'GroupSampler', 'DistributedGroupSampler', 'ConcatDataset', 'CustomDataset', 'CocoDataset', 'GroupSampler', 'DistributedGroupSampler',
'build_dataloader', 'to_tensor', 'random_scale', 'show_ann', 'get_dataset' 'ConcatDataset', 'build_dataloader', 'to_tensor', 'random_scale',
'show_ann', 'get_dataset'
] ]
...@@ -27,4 +27,3 @@ class ConcatDataset(_ConcatDataset): ...@@ -27,4 +27,3 @@ class ConcatDataset(_ConcatDataset):
else: else:
sample_idx = idx - self.cumulative_sizes[dataset_idx - 1] sample_idx = idx - self.cumulative_sizes[dataset_idx - 1]
return dataset_idx, sample_idx return dataset_idx, sample_idx
...@@ -9,6 +9,7 @@ import numpy as np ...@@ -9,6 +9,7 @@ import numpy as np
from .concat_dataset import ConcatDataset from .concat_dataset import ConcatDataset
from .. import datasets from .. import datasets
def to_tensor(data): def to_tensor(data):
"""Convert objects of various python types to :obj:`torch.Tensor`. """Convert objects of various python types to :obj:`torch.Tensor`.
...@@ -72,7 +73,8 @@ def show_ann(coco, img, ann_info): ...@@ -72,7 +73,8 @@ def show_ann(coco, img, ann_info):
def get_dataset(data_cfg): def get_dataset(data_cfg):
if isinstance(data_cfg['ann_file'], list) or isinstance(data_cfg['ann_file'], tuple): if isinstance(data_cfg['ann_file'], list) or \
isinstance(data_cfg['ann_file'], tuple):
ann_files = data_cfg['ann_file'] ann_files = data_cfg['ann_file']
dsets = [] dsets = []
for ann_file in ann_files: for ann_file in ann_files:
......
...@@ -2,7 +2,6 @@ from __future__ import division ...@@ -2,7 +2,6 @@ from __future__ import division
import argparse import argparse
from mmcv import Config from mmcv import Config
from mmcv.runner import obj_from_dict
from mmdet import datasets, __version__ from mmdet import datasets, __version__
from mmdet.apis import (train_detector, init_dist, get_root_logger, from mmdet.apis import (train_detector, init_dist, get_root_logger,
......
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