simplev1_datatset.py

# -*- coding: utf-8 -*-
# ======================================================
# @Time    : 20-12-25 下午9:49
# @Author  : huang ha
# @Email   :
# @File    : simple1_datatset.py
# @Comment: 
# ======================================================

from libv1 import Dataset
import numpy as np


class SimpleV1Dataset(Dataset):
    def __init__(self):
        # 伪造数据
        self.imgs = np.arange(0, 16).reshape(8, 2)

    def __getitem__(self, index):
        return self.imgs[index]

    def __len__(self):
        return self.imgs.shape[0]