diff --git a/ppcls/data/__init__.py b/ppcls/data/__init__.py index de62cc5da0402d93d5f9dab9447c53e5978903b1..661fb3324ed1f3e196287cd5bfbb95d5578ac350 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):