search

vinceecws / SegNet_PyTorch

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

SegNet_PyTorch #1

Closed nargesthy closed 2 years ago

nargesthy commented 2 years ago

I try to run your SegNet script on google Colab. I had multiple issues I fixed, but I got stuck with the last one which is related to the Train_SegNet.py file. The error is as below:

Resume training? (Y/N) Y No checkpoint found at '/content/gdrive/MyDrive/Colab_Notebooks/seg_Narges/SegNet_PyTorch-master/checkpoint.pt' Epoch 1: ^C

Your response is highly appreciated.

vinceecws commented 2 years ago

Hey, the script takes in absolute directories. Try passing in the full path.

dyuming commented 2 years ago

Can you share the camvid dataset you used?

nargesthy commented 2 years ago

Hey, the script takes in absolute directories. Try passing in the full path.

Hi, Thank you for your response. Actually, the issue was solved after running the model on a local GPU, instead of Google Colab. The CamVid dataset is too large to be run on the colab.

nargesthy commented 2 years ago

Can you share the camvid dataset you used?

Here is the link to download the CamVid dataset: http://mi.eng.cam.ac.uk/research/projects/VideoRec/CamVid/

vinceecws commented 2 years ago

Good to hear, closing this issue!