lichen/ieemoo-ai-isempty
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update predict.py.

Browse Source
This commit is contained in:
Brainway
2022-10-18 03:42:41 +00:00
committed by Gitee
parent 55962c44e4
commit fb45a96528
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
predict.py
View File

@ -17,7 +17,7 @@ def parse_args():
parser.add_argument('--split', type=str, default='overlap', help="Split method") # non-overlap
parser.add_argument('--slide_step', type=int, default=2, help="Slide step for overlap split")
parser.add_argument('--smoothing_value', type=float, default=0.0, help="Label smoothing value\n")
parser.add_argument("--pretrained_model", type=str, default="../module/ieemoo-ai-isempty/model/now/emptyjudge5_checkpoint.bin", help="load pretrained model")
parser.add_argument("--pretrained_model", type=str, default="./output/ieemooempty_vit_checkpoint.pth", help="load pretrained model")
return parser.parse_args()
@ -79,8 +79,10 @@ if __name__ == "__main__":
y_true = []
y_pred = []
test_dir = "./emptyJudge5/images/"
dir_dict = {"noemp":"0", "yesemp":"1", "hard": "2", "fly": "3", "stack": "4"}
# test_dir = "./emptyJudge5/images/"
# dir_dict = {"noemp":"0", "yesemp":"1", "hard": "2", "fly": "3", "stack": "4"}
test_dir = "../emptyJudge2/"
dir_dict = {"noempty":"0", "empty":"1"}
total = 0
num = 0
t0 = time.time()