小小顺利的博客

from torch.cuda.amp import autocast, GradScaler # Init Step 1: Create Model model, device, start_epoch = create_model(opt) if torch.cuda.device_count() > 1: model = nn.DataParallel(model) model.cuda() # Init Step 2: Cre...

1.加载模型seg_model=torchvision.models.detection.maskrcnn_resnet50_fpn(pretrained=True)seg_model=seg_model.cuda(1)seg_model=seg_model.eval()2.将图像封装为tensor：torchvision.transforms.functional.to_tensor(img_pil).cuda() #cuda之后可以指定ID3.从模型取出结果：cond_per...