Skip to content

Latest commit

 

History

History

mindspore

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
sparseinst
sparseinst
 
 
README.md
README.md
 
 
dict.py
dict.py
 
 
test.py
test.py
 
 

README.md

SparseInst on MindSpore

Installation

  1. create python 3.8 environment
conda create -n sparseinst-ms python=3.8
  1. activate the new environment
conda activate sparseinst-ms
  1. install mindspore
pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.8.1/MindSpore/gpu/x86_64/cuda-11.1/mindspore_gpu-1.8.1-cp38-cp38-linux_x86_64.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple
  1. install dependencies
pip install mindvision pycocotools opencv-python numpy yacs

Model

We provide the basic SparseInst-R50-GIAM in BaiduPan.

Demo

python test.py --config /path/to/your/checkpoint  --image_name /path/to/your/image --visualize

The results will be saved in ./image_name/