from watermark_verify.inference.rcnn_inference import FasterRCNNInference
from watermark_verify.inference.ssd_inference import SSDInference
from watermark_verify.inference.yolox_inference import YOLOXInference

if __name__ == '__main__':
    # model_filename = "/mnt/e/DevTools/workspace/model_watermark_detect/tests/models/origin/yolox/yolox.onnx"
    # dets = YOLOXInference(model_filename).predict('coco/val2017/000000005992.jpg')
    # model_filename = "/mnt/e/DevTools/workspace/model_watermark_detect/tests/models/whitebox/ssd/ssd.onnx"
    # dets = SSDInference(model_filename).predict('coco/val2017/000000000872.jpg')
    model_filename = "/mnt/e/DevTools/workspace/model_watermark_detect/tests/models/origin/faster-rcnn/faster-rcnn.onnx"
    dets = FasterRCNNInference(model_filename).predict('coco/val2017/000000000872.jpg')
    print(f"dets: {dets}")