docs/si/unet/carvana.html
ඔබටබාගත කිරීමේ උපදෙස් සොයාගත හැකිය Kaggle.
carvana/train ෆෝල්ඩරය තුළ පුහුණු පින්තූර සහ carvana/train_masks ෆෝල්ඩරයේ වෙස් මුහුණු සුරකින්න.
16fromtorchimportnn17frompathlibimportPath1819importtorch.utils.data20importtorchvision.transforms.functional21fromPILimportImage2223fromlabmlimportlab
26classCarvanaDataset(torch.utils.data.Dataset):
image_path රූප සඳහා මාර්ගය වේmask_path වෙස් මුහුණු වෙත යන මාර්ගයයි31def\_\_init\_\_(self,image\_path:Path,mask\_path:Path):
Idමගින් රූප ශබ්ද කෝෂයක් ලබා ගන්න
37self.images={p.stem:pforpinimage\_path.iterdir()}
හැඳුනුම්පතමගින් වෙස් මුහුණු ශබ්ද කෝෂයක් ලබා ගන්න
39self.masks={p.stem[:-5]:pforpinmask\_path.iterdir()}
රූපIDS ලැයිස්තුව
42self.ids=list(self.images.keys())
පරිවර්තනයන්
45self.transforms=torchvision.transforms.Compose([46torchvision.transforms.Resize(572),47torchvision.transforms.ToTensor(),48])
idx රූපයේ දර්ශකය වේ50def\_\_getitem\_\_(self,idx:int):
රූපහැඳුනුම්පත ලබා ගන්න
58id\_=self.ids[idx]
රූපයපටවන්න
60image=Image.open(self.images[id\_])
රූපයපරිවර්තනය කර එය පයිටෝච් ටෙන්සරයක් බවට පරිවර්තනය කරන්න
62image=self.transforms(image)
ආවරණපටවන්න
64mask=Image.open(self.masks[id\_])
වෙස්මුහුණපරිවර්තනය කර එය පයිටෝච් ටෙන්සරයක් බවට පරිවර්තනය කරන්න
66mask=self.transforms(mask)
ආවරණඅගයන් නොතිබුණි 1, එබැවින් අපි එය නිසි පරිදි පරිමාණය කරමු.
69mask=mask/mask.max()
රූපයසහ වෙස්මුහුණ ආපසු ලබා දෙන්න
72returnimage,mask
74def\_\_len\_\_(self):
78returnlen(self.ids)
පරීක්ෂාකිරීමේ කේතය
82if\_\_name\_\_=='\_\_main\_\_':83ds=CarvanaDataset(lab.get\_data\_path()/'carvana'/'train',lab.get\_data\_path()/'carvana'/'train\_masks')