mlp_mixer

MLP-Mixer

Input

Ailia input shape: (1,3,224,224) Range: [0.0, 255.0]

Output

class_count=3
+ idx=0
  category=5[dog ]
  prob=0.9998886585235596
+ idx=1
  category=3[cat ]
  prob=5.7395416661165655e-05
+ idx=2
  category=2[bird ]
  prob=3.886506237904541e-05

Usage

Automatically downloads the onnx and prototxt files on the first run. It is necessary to be connected to the Internet while downloading.

For the sample image,

$ python3 mlp_mixer.py

If you want to specify the input image, put the image path after the --input option.

$ python3 mlp_mixer.py --input IMAGE_PATH

By adding the --video option, you can input the video.
If you pass 0 as an argument to VIDEO_PATH, you can use the webcam input instead of the video file.

$ python3 mlp_mixer.py --video VIDEO_PATH

Reference

MLP-Mixer

Model Format

ONNX opset = 12

Framework

pytorch

Netron

Mixer-B_16.opt.onnx.prototxt

Mixer-L_16.opt.onnx.prototxt