Setup

git clone https://github.com/hxy-0818/ULR2SS.git
cd ULR2SS
conda create -n ULR2SS python=3.8
conda activte ULR2SS
pip install -r requirements.txt

Weight

For convenience, our pre-trained ULR2SS model can be downloaded directly here: ULR2SS_Weight

Demo Test

python inference.py \
  --input  /home/user/ULR2SS/images/rgb_demo1.png \      # path to image/folder
  --output /home/user/ULR2SS/images/output \  # path to save results
  --checkpoint /home/user/ULR2SS/joint_checkpoint_best.pth \ # ckpt path