Commit 7cb703c4 authored by liuhy's avatar liuhy
Browse files

修改test代码

parent 679ae84d
...@@ -75,9 +75,7 @@ def validation(args): ...@@ -75,9 +75,7 @@ def validation(args):
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser(description='parameters to vaildate net') parser = argparse.ArgumentParser(description='parameters to vaildate net')
parser.add_argument('--model', default='model/lprnet.pth', help='model path to vaildate') parser.add_argument('--model', default='model/lprnet.pth', help='model path to vaildate')
# parser.add_argument('--model', default='weights/Final_LPRNet_model.pth', help='model path to vaildate') parser.add_argument('--imgpath', default='imgs', help='the image path')
# parser.add_argument('--imgpath', default='imgs/川JK0707.jpg', help='the image path')
parser.add_argument('--imgpath', default='/code/lpr_ori/data/test', help='the image path')
parser.add_argument('--device', default='cuda', help='Use cuda to vaildate model') parser.add_argument('--device', default='cuda', help='Use cuda to vaildate model')
parser.add_argument('--export_onnx', default=False, help='export model to onnx') parser.add_argument('--export_onnx', default=False, help='export model to onnx')
parser.add_argument('--dynamic', default=False, help='use dynamic batch size') parser.add_argument('--dynamic', default=False, help='use dynamic batch size')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment