Skip to content

Train SegNet #46

Open
Open
@muralabmahmuds

Description

@muralabmahmuds

Hi @zijundeng and everybody,

First of all, thank you for your codes. I am a very beginner in semantic segmentation area.

I want to run your codes using SegNet model, but I didn't find the "train.py" for SegNet. Then, I simply try to modify "train.py" from folder "train/cityscapes-psp_net" and change the net in line 42.
from : net = PSPNet(num_classes=cityscapes.num_classes)
to be : net = SegNet(num_classes=cityscapes.num_classes)

But, I found this error and still cannot solve the problem.
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 31 and 32 in dimension 2 at /opt/conda/conda-bld/pytorch_1525909934016/work/aten/src/THC/generic/THCTensorMath.cu:111

Please, kindly your suggestion.
Thank you.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions