def test():
args = parse_args()
if args.config_file is not None:
cfg_from_file(args.config_file)
test_model()
def test():
test_model()
args = parser.parse_args()
return args
def test():
test_model()
def test_single_image(image_path):
image = cv2.imread(image_path)
test_image(image)
cv2.imwrite('/tmp/np_result.jpg', image)
if __name__ == '__main__':
args = parse_args()
if args.config_file is not None:
cfg_from_file(args.config_file)
if args.onnx_file is not None:
export_onnx_model(args.onnx_file)
elif args.single_image is not None:
test_single_image(args.single_image)
#export_onnx_model("/tmp/bayer_ssd_lite_mbv2.onnx")
#test_single_image('/mnt/500GB/datasets/Bayer_0315/photos/1.jpg')
#test_single_image('/tmp/900x600.jpg')
else:
test_model()