CrackMamba

This repository is the official PyTorch implementation of Mamba meets crack segmentation.

🍓 Paper:

The initial version of the paper can refer to the arXiv version or ResearchGate.

📢 Introduction

This repository mainly includes 2 parts:

Code of CrackMamba
A simple tutorial about how to install Mamba

📌 Updates

2024/07/23: The preprint of our paper is publicly available on arXiv. Link → Arxiv Paper.
2024/07/22: The preprint of our paper is submitted to arXiv.
2024/06/30: This repository is built up! The code will be gradually released.

🏎️ Getting Started

1. Requirements

Our version requirements are lower than those of other Mamba-based research, which is more friendly to those who installed old versions.

Recommended versions are
    * Mamba = 1.1.1 or 1.2.0
    * causal-conv1d = 1.1.1
    * python = 3.8.15
    * pytorch = 1.12.1
    * CUDA 11.6.2 and CUDNN 8.6.0

I want to let you know that the following installation works in Linux (specifically, I used Ubuntu 18.04). It does not work when I try to reproduce it in Win 11. 😤

2. Installation of Mamba

Step 1: Install causal_conv 1.1.1

pip install causal_conv1d==1.1.1

The screenshot of the successful installation is shown below.

Step 2: Install Mamba 1.2.0

cd mamba-1.2.0
python setup.py install

Remark: We have provided the needed Mamba-1.2.0 in this repo for your convenience.
The screenshot of the successful installation is shown below.

Step 3: Install Mamba 1.2.0

hzlbbfrog/CrackMamba