Clone/DIS
1
0
Fork 0
mirror of https://github.com/xuebinqin/DIS.git synced 2024-11-26 08:43:17 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update Inference.py

Browse Source
This commit is contained in:
Xuebin Qin 2022-08-21 12:37:16 -07:00 committed by GitHub
parent d88f41fe51
commit fdb5f91ad1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
IS-Net/Inference.py
View File

@ -17,9 +17,9 @@ from models import *
if __name__ == "__main__":
dataset_path="../datasets/demo_dataset" #Your dataset path
model_path="../saved_models/isnet.pth" #Your model path
result_path="../datasets/demo_result" #The folder path that you want to save the results
dataset_path="../demo_datasets/your_dataset" #Your dataset path
model_path="../saved_models/IS-Net/isnet-general-use.pth" # the model path
result_path="../demo_datasets/your_dataset_result" #The folder path that you want to save the results
input_size=[1024,1024]
net=ISNetDIS()