From 587b2de58cafa55b1bbd3800794fedf2ce1fe266 Mon Sep 17 00:00:00 2001 From: weishengyu Date: Thu, 3 Jun 2021 15:31:18 +0800 Subject: [PATCH] dbg --- ppcls/data/__init__.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/ppcls/data/__init__.py b/ppcls/data/__init__.py index de62cc5d..661fb332 100644 --- a/ppcls/data/__init__.py +++ b/ppcls/data/__init__.py @@ -11,26 +11,25 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import copy import paddle import numpy as np from paddle.io import DistributedBatchSampler, BatchSampler, DataLoader - from ppcls.utils import logger -from . import dataloader -from . import imaug -from . import samplers +from ppcls.data import dataloader +from ppcls.data import imaug # dataset -from .dataloader.imagenet_dataset import ImageNetDataset -from .dataloader.multilabel_dataset import MultiLabelDataset -from .dataloader.common_dataset import create_operators -from .dataloader.vehicle_dataset import CompCars, VeriWild +from ppcls.data.dataloader.imagenet_dataset import ImageNetDataset +from ppcls.data.dataloader.multilabel_dataset import MultiLabelDataset +from ppcls.data.dataloader.common_dataset import create_operators +from ppcls.data.dataloader.vehicle_dataset import CompCars, VeriWild # sampler -from .samplers import DistributedRandomIdentitySampler +from ppcls.data.dataloader import DistributedRandomIdentitySampler -from .preprocess import transform +from ppcls.data.preprocess import transform def build_dataloader(config, mode, device, seed=None): -- GitLab