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

use crossplatform paths instead of unix-style

Browse Source
This commit is contained in:
2trvl 2023-03-28 00:43:04 +03:00
parent f3837183a3
commit 9a188586a3
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
IS-Net/Inference.py
View File

@ -17,9 +17,11 @@ from models import *
if __name__ == "__main__":
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
root_path=os.path.abspath(__file__)
root_path=os.path.dirname(os.path.dirname(root_path))
dataset_path=os.path.join(root_path, "demo_datasets", "your_dataset") # Your dataset path
model_path=os.path.join(root_path, "saved_models", "IS-Net", "isnet-general-use.pth") # the model path
result_path=os.path.join(root_path, "demo_datasets", "your_dataset_result") # The folder path that you want to save the results
input_size=[1024,1024]
net=ISNetDIS()