from rknn.api import RKNN
import os
import cv2
import numpy as np
MODEL = '/mnt/d/Face-Anti-Spoofing-using-DeePixBiS/DeePixBiS.onnx'
RKNN_MODEL = 'DeePixBiS.rknn'


if __name__ == '__main__':
    rknn = RKNN(verbose=True)
    rknn.config(mean_values=[[127.5,127.5,127.5]], std_values=[[127.5,127.5,127.5]],
                        target_platform='rk3588')

    ret = rknn.load_onnx(model=MODEL)
    if ret != 0:
        print('Load model failed!')
        exit(ret)

    ret = rknn.build(do_quantization=False)
    if ret != 0:
        print('Build model failed!')
        exit(ret)
    print('done')

    ret = rknn.export_rknn(RKNN_MODEL)
    if ret != 0:
        print('Export rknn model failed!')
        exit(ret)
    print('done')