Introduction

A repository that has implemented UNet using PyTorch. And this code contains code and examples for visualization.

Environment

• Python - 3.8.2
• Pytorch - 1.12.1
• Matplotlib - 3.7.1

Included Code

• UNet Network
• MNIST Dataset for semantic segmentation
• Segmenation Visualization

Result

• Epoch (0 / 10 / 20)

• Inference by class