search

vinceecws / SegNet_PyTorch

PyTorch implementation of SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation
58 stars 13 forks source link

No Training code in SegNet.py #3

Closed kreuz1995 closed 1 year ago

kreuz1995 commented 1 year ago

There is no class method as Train in SegNet.py. But in Train_SegNet.py, it calls Train from SegNet.py

vinceecws commented 1 year ago

Hello, this must've been accidentally removed during refactoring. I've added the missing class and the training script should work now. Please pull the latest from master.

https://github.com/vinceecws/SegNet_PyTorch/blob/d870acccfeb6d99be38301981e1a5a14735d2d90/CamVid/SegNet.py#L166