深度学习语义分割科研小班第一期
)语义分割的数据集
最重要的语义分割的数据集之一是Pascal VOC2012
(二)代码实现加载数据集
关于数据集可以直接用浏览器下载,也可以通过代码下载
%matplotlib inline
import os
import torch
import torchvision
from d2l import torch as d2l
# 下载数据集
if os.path.isdir('../data/VOCdevkit/'):
voc_dir = "../data/VOCdevkit/VOC2012"
else:
d2l.DATA_HUB['voc2012'] = (d2l.DATA_URL + 'VOCtrainval_11-May-2012.tar',
'4e443f8a2eca6b1dac8a6c57641b67dd40621a49')
voc_dir = d2l.download_extract('voc2012', 'VOCdevkit/VOC2012')