提交 587b2de5 编写于 作者: W weishengyu

dbg

上级 582157f2
......@@ -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):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册