Skip to content

This project involves the segmentation of city images using a Generative Adversarial Network (GAN) based on the Pix2Pix model. The model was trained for 350 epochs with a dataset of 3200 images.

Notifications You must be signed in to change notification settings

Abhigyan126/City-Segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 

Repository files navigation

City Segmentation with GAN Pix2Pix

This project involves the segmentation of city images using a Generative Adversarial Network (GAN) based on the Pix2Pix model. The model was trained for 350 epochs with a dataset of 3200 images.

Screenshot


Screenshot 2024-06-26 at 11 10 08 AM
## Introduction

The goal of this project is to accurately segment images of cityscapes into different classes such as buildings, roads, and sky. We use the Pix2Pix model, a type of conditional GAN, which is well-suited for image-to-image translation tasks.

Dataset

The dataset used for training consists of 3200 images of cityscapes. Each image has corresponding ground truth segmentation masks. The dataset can be found on Kaggle: Cityscapes Image Pairs.

About

This project involves the segmentation of city images using a Generative Adversarial Network (GAN) based on the Pix2Pix model. The model was trained for 350 epochs with a dataset of 3200 images.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published