python深度學(xué)習之多標簽分類器及pytorch實現(xiàn)源碼

更新時間：2022年01月30日 09:14:08 作者：鬼道2022

這篇文章主要為大家介紹了python深度學(xué)習之多標簽分類器的使用說明及pytorch的實現(xiàn)源碼，有需要的朋友可以借鑒參考下，希望能夠有所幫助

多標簽分類器

多標簽分類任務(wù)與多分類任務(wù)有所不同，多分類任務(wù)是將一個實例分到某個類別中，多標簽分類任務(wù)是將某個實例分到多個類別中。多標簽分類任務(wù)有有兩大特點：

類標數(shù)量不確定，有些樣本可能只有一個類標，有些樣本的類標可能高達幾十甚至上百個
類標之間相互依賴，例如包含藍天類標的樣本很大概率上包含白云

如下圖所示，即為一個多標簽分類學(xué)習的一個例子，一張圖片里有多個類別，房子，樹，云等，深度學(xué)習模型需要將其一一分類識別出來。

多標簽分類器損失函數(shù)

代碼實現(xiàn)

針對圖像的多標簽分類器pytorch的簡化代碼實現(xiàn)如下所示。因為圖像的多標簽分類器的數(shù)據(jù)集比較難獲取，所以可以通過對mnist數(shù)據(jù)集中的每個圖片打上特定的多標簽，例如類別1的多標簽可以為[1,1,0,1,0,1,0,0,1]，然后再利用重新打標后的數(shù)據(jù)集訓(xùn)練出一個mnist的多標簽分類器。

from torchvision import datasets, transforms
from torch.utils.data import DataLoader, Dataset
import torch
import torch.nn as nn
import torch.optim as optim
import torch.nn.functional as F
import os
class CNN(nn.Module):
    def __init__(self):
        super().__init__()
        self.Sq1 = nn.Sequential(         
            nn.Conv2d(in_channels=1, out_channels=16, kernel_size=5, stride=1, padding=2),   # (16, 28, 28)                           #  output: (16, 28, 28)
            nn.ReLU(),                    
            nn.MaxPool2d(kernel_size=2),    # (16, 14, 14)
        )
        self.Sq2 = nn.Sequential(
            nn.Conv2d(in_channels=16, out_channels=32, kernel_size=5, stride=1, padding=2),  # (32, 14, 14)
            nn.ReLU(),                      
            nn.MaxPool2d(2),                # (32, 7, 7)
        )
        self.out = nn.Linear(32 * 7 * 7, 100)  
    def forward(self, x):
        x = self.Sq1(x)
        x = self.Sq2(x)
        x = x.view(x.size(0), -1)    
        x = self.out(x)
        ## Sigmoid activation   
        output = F.sigmoid(x)  # 1/(1+e**(-x))
        return output
def loss_fn(pred, target):
    return -(target * torch.log(pred) + (1 - target) * torch.log(1 - pred)).sum()
def multilabel_generate(label):
    Y1 = F.one_hot(label, num_classes = 100)
    Y2 = F.one_hot(label+10, num_classes = 100)
    Y3 = F.one_hot(label+50, num_classes = 100) 	
    multilabel = Y1+Y2+Y3
    return multilabel
        
# def multilabel_generate(label):
# 	multilabel_dict = {}
# 	multi_list = []
# 	for i in range(label.shape[0]):
# 		multi_list.append(multilabel_dict[label[i].item()])
# 	multilabel_tensor = torch.tensor(multi_list)
#     return multilabel
def train():
    epoches = 10
    mnist_net = CNN()
    mnist_net.train()
    opitimizer = optim.SGD(mnist_net.parameters(), lr=0.002)
    mnist_train = datasets.MNIST("mnist-data", train=True, download=True, transform=transforms.ToTensor())
    train_loader = torch.utils.data.DataLoader(mnist_train, batch_size= 128, shuffle=True)
    for epoch in range(epoches):
    	loss = 0 
    	for batch_X, batch_Y in train_loader:
    		opitimizer.zero_grad()
    		outputs = mnist_net(batch_X)
    		loss = loss_fn(outputs, multilabel_generate(batch_Y)) / batch_X.shape[0]
    		loss.backward()
    		opitimizer.step()
    		print(loss)
if __name__ == '__main__':
	train()

以上就是python深度學(xué)習之多標簽分類器及pytorch源碼的詳細內(nèi)容，更多關(guān)于多標簽分類器pytorch源碼的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: