niobures's picture
Segformer++ (code, models, paper)
566e7ce verified

MMSegmentation Setup

Step 0. Prerequisites

  • Pytorch: 2.0.1 (CUDA 11.8)

Step 1. Install MMCV using MIM.

pip install -U openmim
mim install mmengine
mim install "mmcv==2.0.0"

Step 2. Install MMPose

# starting from the project root
cd mmpose
pip install -r requirements.txt
pip install -v -e .

Step 3. Run the SegFormer++ using MMPose