pytorch分割网络数据输入接口

徐梓航

于 2019-11-01 12:54:47 发布

阅读量910

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Pytorch学习文章标签： Pytorch 自定义数据输入分割网络

本文链接：https://blog.youkuaiyun.com/xuyunyunaixuexi/article/details/102855481

本文介绍了如何在PyTorch中利用自定义数据集接口（继承data.Dataset）来处理分割网络的数据输入。作者分享了脚本并验证了其可行性，同时提出疑问：如何根据数据集大小确定crop_size，是否只能通过不断尝试来优化效果，期待读者的讨论和解答。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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.youkuaiyun.com/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)
        self.annotations_base = os.path.join(self.root, 'gtFine_trainvaltest', 'gtFine', self.split)