from MantraNet.mantranet import pre_trained_model, check_forgery

device = "cuda:0"  # to change if you have a GPU with at least 12Go RAM (it will save you a lot of time !)
MantraNetmodel = pre_trained_model(
    weight_path="MantraNet/MantraNetv4.pt", device=device
)
MantraNetmodel.eval()
fig = check_forgery(MantraNetmodel, img_path=r'F:\wql-Graduate\fuxian\IMD-main\images\example4.jpg', device=device)
fig.savefig(r'F:\wql-Graduate\fuxian\IMD-main\images_out\example4_result.jpg')