m0_69562515的博客

原创 图像分割———数据集的建立（pytorch）代码逐行讲解保证你奶奶都会~

针对深度学习，要想训练好一个模型，首先要会建立并读懂数据集的加载，我也看了很多大佬发的讲解版本，接下来针对我的理解给大家解读一下我自己理解的数据集的建立的过程首先大家要知道在pytorch中构建数据集的基本框架:首先第一个init是对我们所用到的一些参数进行初始化，在这个阶段需要加载一些地址，比如图像分割中需要加载img和对应的label；第二个getitem函数，利用index遍历init当中储存的地址，并利用cv2.imshow()或PIL.Image.open()读入图像；第三个函数是len函数..