model_watermark_detect/tests/inference_test.py

import os
import sys
rootpath = str(os.path.abspath(os.path.join(os.path.dirname(__file__), '../')))
sys.path.append(rootpath)
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/origin/ssd/models.onnx"
    # dets = SSDInference(model_filename).predict('VOC2007/JPEGImages/000001.jpg')
    model_filename = "/mnt/e/DevTools/workspace/model_watermark_detect/tests/models/origin/faster-rcnn/model.onnx"
    dets = FasterRCNNInference(model_filename).predict('VOC2007/JPEGImages/000001.jpg')
    print(f"dets: {dets}")