From fdb5f91ad1768a489c129cbaabe3b6a618957c59 Mon Sep 17 00:00:00 2001 From: Xuebin Qin Date: Sun, 21 Aug 2022 12:37:16 -0700 Subject: [PATCH] Update Inference.py --- IS-Net/Inference.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/IS-Net/Inference.py b/IS-Net/Inference.py index 7a8dffd..3e6fae8 100644 --- a/IS-Net/Inference.py +++ b/IS-Net/Inference.py @@ -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()