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 11:49:39 -07:00 committed by GitHub
parent c37bb27b16
commit 1646044b43
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="/home/jiayi.zhu/DIS_projects/demo" #Your dataset path
model_path="/home/jiayi.zhu/DIS_projects/saved_models/IS-Net-New/gpu_itr_654000_traLoss_0.1955_traTarLoss_0.0186_valLoss_0.9414_valTarLoss_0.1418_maxF1_0.9117_mae_0.0388_time_0.030502.pth" #Your model path
result_path="/home/jiayi.zhu/DIS_projects/demo_result" #The folder path that you want to save the results
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
input_size=[1024,1024]
net=ISNetDIS()