pytorch分割网络数据输入接口

 pytorch的自定义接口是真的方便， 记录一下自己分割数据输入的脚本：

# -*- coding: utf-8 -*-
# @Time    : 2019/10/31 21:36
# @Author  : Yunyun Xu
# @Contact : 1443563995@qq.com
# @File    : MyDatasetReader.py
# @Software: Pycharm
# @Blog    : https://me.csdn.net/xuyunyunaixuexi

import os
import numpy as np
import scipy.misc as m
from PIL import Image
from torch.utils import data
from mypath import Path
from torchvision import transforms
import custom_transforms as tr

class MyEggSegmentation(data.Dataset):
    #NUM_CLASSES = 19

    def __init__(self, args, root = Path.db_root_dir("MyEggs"), split = "train"):

        self.root = root
        self.split = split
        self.args = args
        self.image_files = {}
        self.label_files = {}
        #files = {train:[]}

        self.images_base = os.path.join(self.root, 'leftImg8bit', self.split)