Update README.md

fvviz · Feb 1, 2024 · 72972f0 · 72972f0
1 parent d1fa1ff
commit 72972f0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -1,5 +1,7 @@
 <h1> Road segmention UNET </h1>
 
+<img src="{https://img.shields.io/badge/PyTorch-EE4C2C?style=for-the-badge&logo=pytorch&logoColor=white}"/>
+
 Input Image             |  Road mask
 :-------------------------:|:-------------------------:
 ![](https://i.imgur.com/H8Jjrde.jpeg)  |  ![](https://imgur.com/8KoWbce.jpeg)