res mlp pytorch

ResMLP - 파이토치

Facebook AI의 이미지 분류를 위한 모든 MLP 솔루션인 ResMLP를 Pytorch에 구현

설치하다

$ pip install res-mlp-pytorch

용법

 import torch
from res_mlp_pytorch import ResMLP

model = ResMLP (
    image_size = 256 ,
    patch_size = 16 ,
    dim = 512 ,
    depth = 12 ,
    num_classes = 1000
)

img = torch . randn ( 1 , 3 , 256 , 256 )
pred = model ( img ) # (1, 1000)

직사각형 이미지

 import torch
from res_mlp_pytorch import ResMLP

model = ResMLP (
    image_size = ( 128 , 256 ), # (128 x 256)
    patch_size = 16 ,
    dim = 512 ,
    depth = 12 ,
    num_classes = 1000
)

img = torch . randn ( 1 , 3 , 128 , 256 )
pred = model ( img ) # (1, 1000) 

인용

 @misc { touvron2021resmlp ,
    title   = { ResMLP: Feedforward networks for image classification with data-efficient training } , 
    author  = { Hugo Touvron and Piotr Bojanowski and Mathilde Caron and Matthieu Cord and Alaaeldin El-Nouby and Edouard Grave and Armand Joulin and Gabriel Synnaeve and Jakob Verbeek and Hervé Jégou } ,
    year    = { 2021 } ,
    eprint  = { 2105.03404 } ,
    archivePrefix = { arXiv } ,
    primaryClass = { cs.CV }
}