def test_FP32():
print('\nTest FP32 network')
xml_path = os.path.join(os.path.dirname(__file__), '..', '..', 'data', 'candy.xml')
bin_path = os.path.join(os.path.dirname(__file__), '..', '..', 'data', 'candy.bin')
img_path = os.path.join(os.path.dirname(__file__), '..', '..', 'data', 'tram.jpg')
ref_path = os.path.join(os.path.dirname(__file__), '..', '..', 'data', 'tram_candy.png')
model = StyleTransfer(xml_path, bin_path)
img = cv.imread(img_path)
ref = cv.imread(ref_path)
stylized = model.process(img)
assert(cv.norm(stylized, ref, cv.NORM_INF) <= 1)
def test_INT8():
print('\nTest INT8 network')
xml_path = os.path.join(os.path.dirname(__file__), '..', '..', 'data', 'candy_int8.xml')
bin_path = os.path.join(os.path.dirname(__file__), '..', '..', 'data', 'candy_int8.bin')
img_path = os.path.join(os.path.dirname(__file__), '..', '..', 'data', 'tram.jpg')
ref_path = os.path.join(os.path.dirname(__file__), '..', '..', 'data', 'tram_candy_int8.png')
model = StyleTransfer(xml_path, bin_path)
img = cv.imread(img_path)
ref = cv.imread(ref_path)
stylized = model.process(img)
assert(cv.PSNR(stylized, ref) >= 35)