zrichz / dis Goto Github PK

** (2022-Jul.-30)** Thank AK391 for the Web Demo: .

how to run: (1) Clone this repo

git clone https://github.com/xuebinqin/DIS.git

(2) Configuring the environment: go to the DIS/IS-Net folder and run

conda env create -f pytorch18.yml

Or check the requirements.txt to configure dependencies. REMEMBER - LINUX ONLY

(3) activate the conda environment by

conda activate pytorch18

(4) Inference Download optimized model weights isnet-general-use.pth (for general use) from (Google Drive) or (Baidu Pan 提取码:6jh2), and store them in saved_models/IS-Net

(a) Open \ISNet\inference.py and configure your input and output directories (b) Run

python inference.py