GitHunt
GO

GohVh/resnet34-unet

Segmentation model using UNET architecture with ResNet34 as encoder background, designed with PyTorch.

Resnet34-Unet

This segmentation model is an UNET architecture with ResNet34 as encoder background.

๐ŸŒŸ Architecture Diagram

๐Ÿƒ Run

Clone the project

  git clone https://github.com/GohVh/resnet34-unet.git

Open your Jupyter notebook/Google Colab notebook

  %run main.py
  %run predict.py

๐Ÿšฉ Results

Loss

screenshot

Score

screenshot

Accuracy

screenshot

Predicted results

screenshot

screenshot

screenshot

๐Ÿค Contact

๐Ÿ’Ž Acknowledgements

The dataset used in this project is a Semantic Drone Dataset focuses on semantic understanding of urban scenes for increasing the safety of autonomous drone flight and landing procedures. This dataset is prepared by a team of members from Institute of Computer Graphics and Vision, Graz University of Technology.

Below are the journal papers of two architectures used in this project:

Languages

Python100.0%

Contributors

Created December 20, 2022
Updated March 12, 2026
GohVh/resnet34-unet | GitHunt