sparshgarg23 / CamVid_semantic_segmentation

semantic segmentation on camvid using BiSeNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CamVId_Bisenet

Train and test BiseNet architecure on CamVid to get an understanding of new semantic segmentation architectures.

If you are interested in just running the demo,please mention the path to best_dice_loss.pth(it's currently stored in models)

###The architecture was trained on colab for 100 epochs with each epoch taking 2.3 minutes ,suggestions on improving the training process are welcome.

In order to visualize logs please use the following commands

%load_ext tensorboard %tensorboard --logdir runs/

Epoch vs training loss

epoch_loss_epoch_train

Mean IOU

epoch_miou val

Epoch precision

epoch_precision_val

Some results

Demo picture

exp

Result

person_segmentation_100

Ground truth

Seq05VD_f01560_L

Future changes:experiment with more architectures ,train for longer duration and with batch size 16

About

semantic segmentation on camvid using BiSeNet


Languages

Language:Python 100.0%