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

dbg

上级 582157f2
...@@ -11,26 +11,25 @@ ...@@ -11,26 +11,25 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import copy import copy
import paddle import paddle
import numpy as np import numpy as np
from paddle.io import DistributedBatchSampler, BatchSampler, DataLoader from paddle.io import DistributedBatchSampler, BatchSampler, DataLoader
from ppcls.utils import logger from ppcls.utils import logger
from . import dataloader from ppcls.data import dataloader
from . import imaug from ppcls.data import imaug
from . import samplers
# dataset # dataset
from .dataloader.imagenet_dataset import ImageNetDataset from ppcls.data.dataloader.imagenet_dataset import ImageNetDataset
from .dataloader.multilabel_dataset import MultiLabelDataset from ppcls.data.dataloader.multilabel_dataset import MultiLabelDataset
from .dataloader.common_dataset import create_operators from ppcls.data.dataloader.common_dataset import create_operators
from .dataloader.vehicle_dataset import CompCars, VeriWild from ppcls.data.dataloader.vehicle_dataset import CompCars, VeriWild
# sampler # 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): 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.
先完成此消息的编辑!
想要评论请 注册