Skip to content

Commit

Permalink
rename utils.
Browse files Browse the repository at this point in the history
  • Loading branch information
donnyyou committed Jul 27, 2019
1 parent 584a6ba commit 782921b
Show file tree
Hide file tree
Showing 106 changed files with 196 additions and 196 deletions.
2 changes: 1 addition & 1 deletion datasets/cls/data_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
import datasets.tools.transforms as trans
from datasets.cls.loader.default_loader import DefaultLoader
from datasets.tools.collate import collate
from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class DataLoader(object):
Expand Down
4 changes: 2 additions & 2 deletions datasets/cls/loader/default_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,8 @@
import torch.utils.data as data

from exts.tools.parallel import DataContainer
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class DefaultLoader(data.Dataset):
Expand Down
2 changes: 1 addition & 1 deletion datasets/det/data_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
from datasets.det.loader.fasterrcnn_loader import FasterRCNNLoader
from datasets.det.loader.default_loader import DefaultLoader
from datasets.tools.collate import collate
from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class DataLoader(object):
Expand Down
6 changes: 3 additions & 3 deletions datasets/det/loader/default_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,9 @@
import torch.utils.data as data

from exts.tools.parallel import DataContainer
from util.helpers.json_helper import JsonHelper
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.json_helper import JsonHelper
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log

# torch.multiprocessing.set_sharing_strategy('file_system')

Expand Down
6 changes: 3 additions & 3 deletions datasets/det/loader/fasterrcnn_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,9 @@
import torch.utils.data as data

from exts.tools.parallel import DataContainer
from util.helpers.json_helper import JsonHelper
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.json_helper import JsonHelper
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class FasterRCNNLoader(data.Dataset):
Expand Down
2 changes: 1 addition & 1 deletion datasets/gan/data_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
import datasets.tools.cv2_aug_transforms as cv2_aug_trans
import datasets.tools.transforms as trans
from datasets.tools.collate import collate
from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class DataLoader(object):
Expand Down
2 changes: 1 addition & 1 deletion datasets/gan/loader/cyclegan_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import torch.utils.data as data

from exts.tools.parallel import DataContainer
from util.helpers.image_helper import ImageHelper
from tools.helper.image_helper import ImageHelper


class CycleGANLoader(data.Dataset):
Expand Down
4 changes: 2 additions & 2 deletions datasets/gan/loader/default_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
import torch.utils.data as data

from exts.tools.parallel import DataContainer
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class DefaultLoader(data.Dataset):
Expand Down
4 changes: 2 additions & 2 deletions datasets/gan/loader/facegan_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,8 @@
import torch.utils.data as data

from exts.tools.parallel import DataContainer
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class FaceGANLoader(data.Dataset):
Expand Down
6 changes: 3 additions & 3 deletions datasets/gan/preprocess/face/face_alignmentor.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@
# https://github.com/1adrianb/face-alignment pip install face_alignment
import face_alignment as fa

from util.helpers.file_helper import FileHelper
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.file_helper import FileHelper
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class FaceAlignmentor(object):
Expand Down
2 changes: 1 addition & 1 deletion datasets/ins/data_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
import datasets.tools.cv2_aug_transforms as cv2_aug_trans
import datasets.tools.transforms as trans
from datasets.tools.collate import collate
from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class DataLoader(object):
Expand Down
6 changes: 3 additions & 3 deletions datasets/ins/loader/default_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@
from torch.utils import data

from exts.tools.parallel import DataContainer
from util.helpers.image_helper import ImageHelper
from util.helpers.json_helper import JsonHelper
from util.tools.logger import Logger as Log
from tools.helper.image_helper import ImageHelper
from tools.helper.json_helper import JsonHelper
from tools.util.logger import Logger as Log


class DefaultLoader(data.Dataset):
Expand Down
2 changes: 1 addition & 1 deletion datasets/pose/data_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
import datasets.tools.cv2_aug_transforms as cv2_aug_trans
import datasets.tools.transforms as trans
from datasets.tools.collate import collate
from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class DataLoader(object):
Expand Down
6 changes: 3 additions & 3 deletions datasets/pose/loader/default_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,9 @@

from exts.tools.parallel import DataContainer
from datasets.pose.utils.heatmap_generator import HeatmapGenerator
from util.helpers.json_helper import JsonHelper
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.json_helper import JsonHelper
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class DefaultLoader(data.Dataset):
Expand Down
6 changes: 3 additions & 3 deletions datasets/pose/loader/openpose_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,9 +12,9 @@
from exts.tools.parallel import DataContainer
from datasets.pose.utils.heatmap_generator import HeatmapGenerator
from datasets.pose.utils.paf_generator import PafGenerator
from util.helpers.json_helper import JsonHelper
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.json_helper import JsonHelper
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class OpenPoseLoader(data.Dataset):
Expand Down
2 changes: 1 addition & 1 deletion datasets/pose/utils/heatmap_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import numpy as np
import torch

from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class HeatmapGenerator(object):
Expand Down
2 changes: 1 addition & 1 deletion datasets/seg/data_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
import datasets.tools.cv2_aug_transforms as cv2_aug_trans
import datasets.tools.transforms as trans
from datasets.tools.collate import collate
from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class DataLoader(object):
Expand Down
4 changes: 2 additions & 2 deletions datasets/seg/loader/default_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
from torch.utils import data

from exts.tools.parallel import DataContainer
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class DefaultLoader(data.Dataset):
Expand Down
6 changes: 3 additions & 3 deletions datasets/test/loader/default_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@
import torch.utils.data as data

from exts.tools.parallel import DataContainer
from util.helpers.file_helper import FileHelper
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.file_helper import FileHelper
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class DefaultLoader(data.Dataset):
Expand Down
6 changes: 3 additions & 3 deletions datasets/test/loader/facegan_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@
import torch.utils.data as data

from exts.tools.parallel import DataContainer
from util.helpers.json_helper import JsonHelper
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.json_helper import JsonHelper
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class FaceGANLoader(data.Dataset):
Expand Down
6 changes: 3 additions & 3 deletions datasets/test/loader/json_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@
import torch.utils.data as data

from exts.tools.parallel import DataContainer
from util.helpers.json_helper import JsonHelper
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.json_helper import JsonHelper
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class JsonLoader(data.Dataset):
Expand Down
4 changes: 2 additions & 2 deletions datasets/test/loader/list_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
import torch.utils.data as data

from exts.tools.parallel import DataContainer
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log


class ListLoader(data.Dataset):
Expand Down
2 changes: 1 addition & 1 deletion datasets/test/test_data_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
from datasets.tools.transforms import ToTensor, Normalize, Compose
import datasets.tools.pil_aug_transforms as pil_aug_trans
import datasets.tools.cv2_aug_transforms as cv2_aug_trans
from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class TestDataLoader(object):
Expand Down
4 changes: 2 additions & 2 deletions datasets/tools/collate.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,8 @@
from torch._six import string_classes, int_classes

from exts.tools.parallel import DataContainer
from util.helpers.tensor_helper import TensorHelper
from util.tools.logger import Logger as Log
from tools.helper.tensor_helper import TensorHelper
from tools.util.logger import Logger as Log


def stack(batch, data_key=None, device_ids=None):
Expand Down
2 changes: 1 addition & 1 deletion datasets/tools/transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
import torch
from PIL import Image

from util.helpers.image_helper import ImageHelper
from tools.helper.image_helper import ImageHelper


class Normalize(object):
Expand Down
4 changes: 2 additions & 2 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,8 @@

from runner.method_selector import MethodSelector
from runner.tools.controller import Controller
from util.tools.configer import Configer
from util.tools.logger import Logger as Log
from tools.util.configer import Configer
from tools.util.logger import Logger as Log


def str2bool(v):
Expand Down
2 changes: 1 addition & 1 deletion metric/cls/cls_running_score.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
# Image classification running score.


from util.tools.average_meter import AverageMeter
from tools.util.average_meter import AverageMeter


class ClsRunningScore(object):
Expand Down
4 changes: 2 additions & 2 deletions metric/det/coco_evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,8 @@

from pycocotools.coco import COCO
from pycocotools.cocoeval import COCOeval
from util.tools.configer import Configer
from util.tools.logger import Logger as Log
from tools.util.configer import Configer
from tools.util.logger import Logger as Log


class CocoEvaluator(object):
Expand Down
4 changes: 2 additions & 2 deletions metric/det/voc_evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,8 @@
else:
import xml.etree.ElementTree as ET

from util.tools.configer import Configer
from util.tools.logger import Logger as Log
from tools.util.configer import Configer
from tools.util.logger import Logger as Log


class VOCEvaluator(object):
Expand Down
4 changes: 2 additions & 2 deletions metric/pose/coco_evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,8 @@

from pycocotools.coco import COCO
from pycocotools.cocoeval import COCOeval
from util.tools.configer import Configer
from util.tools.logger import Logger as Log
from tools.util.configer import Configer
from tools.util.logger import Logger as Log


class CocoEvaluator(object):
Expand Down
6 changes: 3 additions & 3 deletions metric/seg/ade20k_evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@
import argparse

from metric.seg.seg_running_score import SegRunningScore
from util.helpers.image_helper import ImageHelper
from util.tools.logger import Logger as Log
from util.tools.configer import Configer
from tools.helper.image_helper import ImageHelper
from tools.util.logger import Logger as Log
from tools.util.configer import Configer


class ADE20KEvaluator(object):
Expand Down
6 changes: 3 additions & 3 deletions metric/seg/coco_evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,9 @@
from pycocotools.cocoeval import COCOeval
from pycocotools import mask as maskUtils

from util.tools.configer import Configer
from util.helpers.mask_helper import MaskHelper
from util.tools.logger import Logger as Log
from tools.util.configer import Configer
from tools.helper.mask_helper import MaskHelper
from tools.util.logger import Logger as Log


class CocoEvaluator(object):
Expand Down
2 changes: 1 addition & 1 deletion model/backbones/backbone_selector.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
from model.backbones.mobilenet.mobilenet_backbone import MobileNetBackbone
from model.backbones.densenet.densenet_backbone import DenseNetBackbone
from model.backbones.squeezenet.squeezenet_backbone import SqueezeNetBackbone
from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class BackboneSelector(object):
Expand Down
2 changes: 1 addition & 1 deletion model/cls/model_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from model.cls.nets.vgg import VGG
from model.cls.nets.shufflenetv2 import ShuffleNetV2, ShuffleResNetV2
from model.cls.loss.cls_modules import ICCELoss, ICCenterLoss
from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log

CLS_MODEL_DICT = {
'vgg11': VGG,
Expand Down
2 changes: 1 addition & 1 deletion model/cls/nets/densenet.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
import torch.nn.functional as F
from torch.autograd import Variable

from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class Bottleneck(nn.Module):
Expand Down
2 changes: 1 addition & 1 deletion model/cls/nets/dpn.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import torch.nn.functional as F
from torch.autograd import Variable

from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class Bottleneck(nn.Module):
Expand Down
2 changes: 1 addition & 1 deletion model/cls/nets/resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import torch.nn.functional as F
from torch.autograd import Variable

from util.tools.logger import Logger as Log
from tools.util.logger import Logger as Log


class BasicBlock(nn.Module):
Expand Down
Loading

0 comments on commit 782921b

Please sign in to comment.