-->

Tutorial Deep Learning : Digit Recognetion menggunakan Pytorch CNN

 


Tutorial Deep Learning : Digit Recognetion menggunakan Pytorch CNN

        Pada Artikel ini kita akan belajar membuat program CNN Deep Learning " Digit Recognetion

Perlu kita ketahui Convolutional neural network  merupakan model baru dibidang pengenalan objek, CNN memiliki layer konvolusi dan layer pooling yang memungkinkan proses pembelajaran fitur secara hirarki dari data. Dataset yang digunakan berasal dari kaggle, telah disiapkan unduh_data 

 

Mempersiapkan Dataset

    Pada langkah ini, melakukan Pre-Processing. 


class DatasetMNIST(torch.utils.data.Dataset):
    def __init__(selfdatatransform=None):
        self.data = data
        self.transform = transform
        
    def __len__(self):
        return len(self.data)
    
    def __getitem__(selfindex):
        item = self.data.iloc[index]
                
        gambar = item[1:].values.astype(np.uint8).reshape((2828))
        label = item[0]
        
        if self.transform is not None:
            gambar = self.transform(gambar)
            
        return gambar, label
 
 
BATCH_SIZE = 100
VALID_SIZE = 0.15 

transform_train = transforms.Compose([
    transforms.ToPILImage(),
   
    transforms.ToTensor(),
    transforms.Normalize(mean=(0.5,), std=(0.5,))
])

transform_valid = transforms.Compose([
    transforms.ToPILImage(),
    transforms.ToTensor(),
    transforms.Normalize(mean=(0.5,), std=(0.5,))
])
 



 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

  

LihatTutupKomentar

a