Skip to content

Liangbo-Cheng/mamba_capsule

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

34 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

πŸ’Š Mamba Capsule Routing Towards Part-Whole Relational Camouflaged Object Detection (MCRNet)

MCRNet

We introduce the Mamba to generate type-level mamba capsules from the pixel-level capsules for routing, which ensures a lightweight computation, further exploring the part-whole hierarchical relationships in COD.


πŸ“Œ Environmental Setups

To set up your environment and install dependencies, run the following commands:

# Create virtual environment
conda create -n mcr python=3.10
conda activate mcr
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia
cd MCRNet
pip install -r requirements.txt

πŸ“Œ Data Preparation

Download the camouflaged object detection datasets from Baidu, you can put datasets into the folder ./data/'. PIN: ss04

πŸ“Œ Checkpoints

We offer the training weights of our MCRNet model on Baidu, which should be put into ./checkpoint/'. PIN: cs28

πŸ“Œ Results

The prediction maps of our MCRNet can be found on Baidu. PIN: l27b

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages