Skip to content

Latest commit

 

History

History
15 lines (11 loc) · 1.04 KB

File metadata and controls

15 lines (11 loc) · 1.04 KB

Dilation-Pytorch-Semantic Segmentation

A PyTorch implementation of semantic segmentation according to Multi-Scale Context Aggregation by Dilated Convolutions by Yu and Koltun. Pretrained weights are obtained from dialation-Tensorflow and thanks to Andrea Palazzi for his work from which this work has been inspired.

Input

Output

How To

  1. Download the pretrained weights from cityscapes
  2. Keep that in the data folder
  3. Run CAN_network.py for the demo.

Notes I have kept the output image size 1024x1024x3 due to memory constraint. If you want the size to be 1024x2048x3, have enough memory and change the sizes appropriately in the code.