/GCDSS

The official code implementation of Generalized Category Discovery in Semantic Segmentation

Primary LanguageJupyter Notebook

GCDSS - Generalized Category Discovery in Semantic Segmentation

Welcome to the official repository for the Generalized Category Discovery in Semantic Segmentation (GCDSS) project!

Overview

This repository contains the code implementation for GCDSS. We are currently in the process of organizing and refining the codebase to make it more user-friendly. Currently, this is a preview version, and continuous optimization of the code will follow. This work is led by Dr. Xin Tan (ECNU).

Usage Instructions

Preparing the Dataset

  1. Place the Cityscapes dataset in the following directory:

    datasets/Cityscapes
    
  2. Run the preprocessing script:

    python preprocess.py

Running the Main Program

  • After preprocessing the data, you can run the main script:
    python main.py

To-Do

  1. Organize and refine the self-training related code to enhance clarity and functionality.
  2. Further streamline the code to make it more user-friendly.

Contact

For inquiries or further information, contact: pengzhengyuan@sjtu.edu.cn

Happy coding!