Giter VIP home page Giter VIP logo

mlp-mixer-tf2's Introduction

[TensorFlow 2] MLP-Mixer: An all-MLP Architecture for Vision

TensorFlow implementation of "MLP-Mixer: An all-MLP Architecture for Vision"

Concept

The MLP-Mixer architecture [1].

Patch Embedding

Mixer Block

MLP in Mixer Blcok

Results

Loss

Performance

Indicator Value
Accuracy 0.97860
Precision 0.97882
Recall 0.97808
F1-Score 0.97829
Confusion Matrix
[[ 968    0    0    0    2    0    3    0    4    3]
 [   0 1127    0    1    1    0    2    1    3    0]
 [   3    0 1002    6    6    0    1    5    9    0]
 [   1    0    2  993    0    0    0    2   10    2]
 [   0    1    0    0  974    0    3    0    1    3]
 [   3    0    0   17    4  839    9    2   12    6]
 [   4    2    0    0   10    0  940    0    2    0]
 [   0    3    4    4    5    0    0 1005    3    4]
 [   0    0    3    1    5    0    1    3  956    5]
 [   1    3    0    3   14    1    0    4    1  982]]
Class-0 | Precision: 0.98776, Recall: 0.98776, F1-Score: 0.98776
Class-1 | Precision: 0.99208, Recall: 0.99295, F1-Score: 0.99251
Class-2 | Precision: 0.99110, Recall: 0.97093, F1-Score: 0.98091
Class-3 | Precision: 0.96878, Recall: 0.98317, F1-Score: 0.97592
Class-4 | Precision: 0.95397, Recall: 0.99185, F1-Score: 0.97254
Class-5 | Precision: 0.99881, Recall: 0.94058, F1-Score: 0.96882
Class-6 | Precision: 0.98019, Recall: 0.98121, F1-Score: 0.98070
Class-7 | Precision: 0.98337, Recall: 0.97763, F1-Score: 0.98049
Class-8 | Precision: 0.95504, Recall: 0.98152, F1-Score: 0.96810
Class-9 | Precision: 0.97711, Recall: 0.97324, F1-Score: 0.97517

Total | Accuracy: 0.97860, Precision: 0.97882, Recall: 0.97808, F1-Score: 0.97829

Requirements

Reference

[1] Tolstikhin, Ilya O., et al. "Mlp-mixer: An all-mlp architecture for vision." Advances in Neural Information Processing Systems 34 (2021).

mlp-mixer-tf2's People

Contributors

yeonghyeon avatar

Stargazers

 avatar  avatar  avatar  avatar

Watchers

 avatar  avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.