Tramac / Fast-SCNN-pytorch

A PyTorch Implementation of Fast-SCNN: Fast Semantic Segmentation Network

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

作者你好,我想问一下如何训练自己的数据集

Liu6697 opened this issue · comments

你好,我想训练一下aeroscapes数据集,他和pascal voc很像,所以我用了您另外一个project中的pascal_voc.py,我在运行之后,出现了如下问题:

Traceback (most recent call last):
File "train.py", line 211, in
trainer.train()
File "train.py", line 138, in train
for i, (images, targets) in enumerate(self.train_loader):
ValueError: too many values to unpack (expected 2)

我感觉他应该是和图像的大小有关系,因为我用的数据集的图像是1280x720,我想问一下我该如何转换图片的大小使其于网络匹配?