from torchvision import transformsimport torchimport cv2class ScaleImageobject def __call__self image return image - 1275128norm_images = transformsCompose ScaleImage def transimgs
There is a mistake in the code, the transform method of ImagePreprocess class is missing self parameter in its definition
class ImagePreprocess: def init(self, device="cpu"): self.norm_image = transforms.Compose([ScaleImage()]) self.device = device
def transform(self, imgs, img_size):
imgs = [cv2.resize(img, (img_size, img_size)) for img in imgs]
imgs = np.asarray(imgs)
imgs = np.transpose(np.copy(imgs[:,:,:,::-1]), (0, 3, 1, 2)).astype(np.float32)
imgs_tensor = torch.from_numpy(imgs).to(self.device)
imgs_norm = self.norm_image(imgs_tensor)
return imgs_norm.cpu().detach().numpy()
原文地址: http://www.cveoy.top/t/topic/6oT 著作权归作者所有。请勿转载和采集!