bayueyaoshao

from models.unet import UNet import torch use_cuda = torch.cuda.is_available() device = torch.device('cuda' if use_cuda else 'cpu') model = UNet().to(device) model_path = r"D:\Work\doc\RIAD\best.pt" export_path = r"D:\Work\doc\RIAD\RIAD_model_320.onnx" check_point = torch.load(model_path, map_location=device) model.load_state_dict(check_point['model'])...

The input images and target masks should be in the data/imgs and data/masks folders respectively (note that the imgs and masks folder should not contain any sub-folder or any other...