from watermark_verify.inference.rcnn import predict_and_detect if __name__ == '__main__': # detect_result = predict_and_detect('trigger/images/2/street.jpg', 'faster-rcnn.onnx', 'trigger/qrcode_positions.txt', (600, 600)) detect_result = predict_and_detect('trigger/images/0/000000000030.jpg', 'faster-rcnn.onnx', 'trigger/qrcode_positions.txt', (600, 600)) print(detect_result)