PINTO_model_zoo

PINTO_model_zoo

Please read the contents of the LICENSE file located directly under each folder before using the model. My model conversion scripts are released under the MIT license, but the license of the source model itself is subject to the license of the provider repository.

Contributors

Made with contrib.rocks.

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

TensorFlow Lite, OpenVINO, CoreML, TensorFlow.js, TF-TRT, MediaPipe, ONNX [.tflite, .h5, .pb, saved_model, tfjs, tftrt, mlmodel, .xml/.bin, .onnx]

I have been working on quantization of various models as a hobby, but I have skipped the work of making sample code to check the operation because it takes a lot of time. I welcome a pull request from volunteers to provide sample code. :smile:

[Note Jan 05, 2020] Currently, the MobileNetV3 backbone model and the Full Integer Quantization model do not return correctly.

[Note Jan 08, 2020] If you want the best performance with RaspberryPi4/3, install Ubuntu 19.10 aarch64 (64bit) instead of Raspbian armv7l (32bit). The official Tensorflow Lite is performance tuned for aarch64. On aarch64 OS, performance is about 4 times higher than on armv7l OS.

My article

• [Japanese ver.] [Tensorflow Lite] Various Neural Network Model quantization methods for Tensorflow Lite (Weight Quantization, Integer Quantization, Full Integer Quantization, Float16 Quantization, EdgeTPU). As of May 05, 2020.

• [English ver.] [Tensorflow Lite] Various Neural Network Model quantization methods for Tensorflow Lite (Weight Quantization, Integer Quantization, Full Integer Quantization, Float16 Quantization, EdgeTPU). As of May 05, 2020.

• Conversion of PyTorch->ONNX->OpenVINO IR model to Tensorflow saved_model / h5 / tflite / pb

• [English] Converting PyTorch, ONNX, Caffe, and OpenVINO (NCHW) models to Tensorflow / TensorflowLite (NHWC) in a snap - Qiita

• [TF2 Object Detection] Converting SSD models into .tflite uint8 format #9371

• tf.image.resizeを含むFull Integer Quantization (.tflite)モデルのEdgeTPUモデルへの変換後の推論時に発生する "main.ERROR - Only float32 and uint8 are supported currently, got -xxx.Node number n (op name) failed to invoke" エラーの回避方法

• A standalone 2MB installer for TensorflowLite v2.4.0-rc4 and a libedgetpu.so.1 build intended to run on a Yocto-generated environment

• [Japanese] Custom Operation入りのtfliteを逆コンバートしてJSON化し標準OPへ置き換えたうえでtfliteを再生成する方法

• Generate saved_model, tfjs, tf-trt, EdgeTPU, CoreML, quantized tflite, ONNX, OpenVINO, Myriad Inference Engine blob and .pb from .tflite.

• Add a custom OP to the TFLite runtime to build the whl installer (for Python), MaxPoolingWithArgmax2D, MaxUnpooling2D, Convolution2DTransposeBias

• ONNX to JSON

• JSON to ONNX

• Steps to merge two ONNX files into one

• Inverse Quantization of tflite's Sparse Tensor Densify to Refine a Clean Float32 Model

• Replace PyTorch's argsort with sort and export to ONNX

• A very simple tool that compresses the overall size of the ONNX model by aggregating duplicate constant values as much as possible. Simple Constant value Shrink for ONNX.

• Simple node deletion tool for onnx. I only test very miscellaneous and limited patterns as a hobby.

• A very simple tool for situations where optimization with onnx-simplifier would exceed the Protocol Buffers upper file size limit of 2GB, or simply to separate onnx files to any size you want. Simple Network Extraction for ONNX.

• Simple tool to combine onnx models. Simple Network Combine Tool for ONNX.

• A set of simple tools for splitting, merging, OP deletion, size compression, rewriting attributes and constants, OP generation, and change opset for ONNX models.

  

• Self-Created Tools to convert ONNX files (NCHW) to TensorFlow format (NHWC). The purpose of this tool is to solve the massive Transpose extrapolation problem in onnx-tensorflow (onnx-tf).

  

List of pre-quantized models

* WQ = Weight Quantization ** OV = OpenVINO IR *** CM = CoreML **** DQ = Dynamic Range Quantization

1. Image Classification

No.	Model Name	Link	FP32	FP16	INT8	DQ	TPU	WQ	OV	CM	TFJS	TF-TRT	ONNX	Remarks
004	Efficientnet	■■■	⚫		⚫			⚫
010	Mobilenetv3	■■■	⚫		⚫			⚫
011	Mobilenetv2	■■■	⚫		⚫	⚫		⚫
016	Efficientnet-lite	■■■	⚫	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫
070	age-gender-recognition	■■■	⚫	⚫	⚫			⚫	⚫	⚫	⚫	⚫	⚫
083	Person_Reidentification	■■■	⚫	⚫	⚫			⚫	⚫	⚫	⚫	⚫	⚫	248,277,286,287,288,300
087	DeepSort	■■■	⚫	⚫	⚫			⚫	⚫	⚫		⚫	⚫
124	person-attributes-recognition-crossroad-0230	■■■	⚫	⚫	⚫			⚫	⚫	⚫	⚫	⚫	⚫
125	person-attributes-recognition-crossroad-0234	■■■	⚫	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫
126	person-attributes-recognition-crossroad-0238	■■■	⚫	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫
175	face-recognition-resnet100-arcface-onnx	■■■	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫	⚫	RGB/BGR,112x112,[1,512]
187	vehicle-attributes-recognition-barrier-0039	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	72x72
188	vehicle-attributes-recognition-barrier-0042	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	72x72
191	anti-spoof-mn3	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫		⚫	128x128
192	open-closed-eye-0001	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫		⚫	32x32
194	face_recognizer_fast	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	112x112
195	person_reid_youtu	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	256x128, ReID
199	NSFW	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	224x224
244	FINNger	■■■	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫		⚫	96x96
256	SFace	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫		⚫	112x112
257	PiCANet	■■■							⚫				⚫	BDDA,SAGE/224x224
259	Emotion_FERPlus	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫		⚫	64x64
290	AdaFace	■■■	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫		⚫	112x112
317	MobileOne	■■■											⚫	224x224
346	facial_expression_recognition_mobilefacenet	■■■	⚫	⚫	⚫	⚫		⚫					⚫	112x112
379	PP-LCNetV2	■■■	⚫	⚫									⚫	224x224
429	OSNet	■■■											⚫	256x128, ReID
430	FastReID	■■■											⚫	384x128, ReID
431	NITEC	■■■											⚫	224x224, Gaze Estimation
432	face-reidentification-retail-0095	■■■	⚫	⚫									⚫	128x128, FaceReID
451	DAN	■■■											⚫	224x224, Facial Expression
452	FairFace	■■■											⚫	224x224, Face Attribute
453	FairDAN	■■■											⚫	224x224, Face Attribute + Facial Expression
462	Gaze-LLE	■■■											⚫	448x448, Attention
474	Gaze-LLE-DINOv3	■■■											⚫	640x640,416x416,320x320, Attention
475	VSDLM	■■■											⚫	30x48, Lip motion
476	OCEC	■■■											⚫	24x40, Wink/Blink
477	PGC	■■■											⚫	32x32, Pointing
478	SC	■■■											⚫	32x24, Sitting
479	PUC	■■■											⚫	32x24, Phone Usage Classifier
480	HSC	■■■											⚫	48x48, Happy smile classifier
481	WHC	■■■											⚫	4x32x32,6x32x32,8x32x32, Waving Hand Classification
483	LVFace	■■■											⚫	Nx3x112x112, Cosine similarity
484	TransFace	■■■											⚫	Nx3x112x112, Cosine similarity
486	MWC	■■■											⚫	Nx3x48x48, Mask wearing classifier
489	Glasses-Detector	■■■											⚫	Glasses
491	SGC	■■■											⚫	Sunglasses
492	HHC	■■■											⚫	Hat
494	BPC	■■■											⚫	Background Plain classification
495	Comprehensive-Head-Classification	■■■	⚫										⚫	Comprehensive head classification
497	PPC	■■■											⚫	Binary classification to determine whether the subject is holding a smartphone
502	PersonViT	■■■											⚫	Person Re-Identification
503	CDNet	■■■											⚫	Person Re-Identification

2. 2D Object Detection

No.	Model Name	Link	FP32	FP16	INT8	TPU	DQ	WQ	OV	CM	TFJS	TF-TRT	ONNX	Remarks
002	Mobilenetv3-SSD	■■■	⚫		⚫			⚫		⚫
006	Mobilenetv2-SSDlite	■■■	⚫		⚫	⚫		⚫
008	Mask_RCNN_Inceptionv2	■■■	⚫	⚫				⚫
018	EfficientDet	■■■	⚫	⚫	⚫			⚫	⚫	⚫	⚫	⚫	⚫
023	Yolov3-nano	■■■	⚫	⚫				⚫					⚫
024	Yolov3-lite	■■■	⚫	⚫	⚫			⚫	⚫				⚫
031	Yolov4	■■■	⚫	⚫	⚫	⚫		⚫					⚫
034	SSD_Mobilenetv2_mnasfpn	■■■	⚫	⚫	⚫	⚫		⚫					⚫
038	SSDlite_MobileDet_edgetpu	■■■	⚫	⚫	⚫	⚫		⚫			⚫	⚫	⚫
039	SSDlite_MobileDet_cpu	■■■	⚫	⚫	⚫			⚫
042	Centernet	■■■	⚫	⚫	⚫			⚫	⚫	⚫	⚫	⚫	⚫
045	SSD_Mobilenetv2_oid_v4	■■■	⚫	⚫	⚫	⚫		⚫			⚫
046	Yolov4-tiny	■■■	⚫	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫
047	SpineNetMB_49	■■■	⚫	⚫	⚫			⚫					⚫	Mobile RetinaNet
051	East_Text_Detection	■■■	⚫	⚫	⚫	⚫		⚫	⚫		⚫	⚫	⚫
054	KNIFT	■■■	⚫	⚫	⚫			⚫	⚫				⚫	MediaPipe
056	TextBoxes++ with dense blocks, separable convolution and Focal Loss	■■■	⚫	⚫	⚫			⚫	⚫	⚫		⚫	⚫
058	keras-retinanet	■■■	⚫	⚫				⚫	⚫		⚫	⚫	⚫	resnet50_coco_best_v2.1.0.h5,320x320
072	NanoDet	■■■	⚫	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫	issue #274
073	RetinaNet	■■■	⚫	⚫	⚫			⚫	⚫	⚫	⚫	⚫	⚫
074	Yolact	■■■	⚫	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫
085	Yolact_Edge	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	21/10/05 new MobileNetV2(550x550)
089	DETR	■■■	⚫	⚫	⚫			⚫					⚫	256x256
103	EfficientDet_lite	■■■	⚫	⚫	⚫	⚫		⚫	⚫		⚫	⚫	⚫	lite0,lite1,lite2,lite3,lite4
116	DroNet	■■■	⚫	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫	DroNet,DroNetV3
123	YOLOR	■■■	⚫	⚫	⚫	⚫		⚫	⚫	⚫	⚫		⚫	ssss_s2d/320x320,640x640,960x960,1280x1280
132	YOLOX	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫		⚫	nano,tiny,s,m,l,x/256x320,320x320,416x416,480x640,544x960,736x1280,1088x1920
143	RAPiD	■■■	⚫	⚫	⚫			⚫	⚫				⚫	Fisheye, cepdof/habbof/mw_r, 608x608/1024x1024
145	text_detection_db	■■■	⚫	⚫	⚫			⚫	⚫	⚫	⚫	⚫	⚫	480x640
151	object_detection_mobile_object_localizer	■■■	⚫	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫	192x192
169	spaghettinet_edgetpu	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫		⚫	⚫	⚫	320x320,S/M/L
174	PP-PicoDet	■■■	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫	⚫	S/M/L,320x320/416x416/640x640
178	vehicle-detection-0200	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	256x256,PriorBoxClustered->ndarray(0.npy)
179	person-detection-0202	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	512x512,PriorBoxClustered->ndarray(0.npy)
183	pedestrian-detection-adas-0002	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	384x672,PriorBox->ndarray(0.npy)
184	pedestrian-and-vehicle-detector-adas-0001	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	384x672,PriorBox->ndarray(0.npy)
185	person-vehicle-bike-detection-crossroad-0078	■■■	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫	⚫	1024x1024,PriorBoxClustered->ndarray(0.npy)
186	person-vehicle-bike-detection-crossroad-1016	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	512x512,PriorBoxClustered->ndarray(0.npy)
189	vehicle-license-plate-detection-barrier-0106	■■■	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	⚫	300x300,PriorBoxClustered->ndarray(0.npy)
190	person-detection-asl-0001	■■■	⚫	⚫	⚫		⚫	⚫	⚫				⚫	320x320
197	yolact-resnet50-fpn	■■■	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫	⚫	RGB,550x550
198	YOLOF	■■■	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫	⚫	⚫	BGR/RGB,608x608
221	YOLACT-PyTorch	■■■	⚫	⚫	⚫		⚫	⚫	⚫		⚫	⚫	⚫	180x320,240x320,320x480,480x640,544x544,720x1280
226	CascadeTableNet	■■■	⚫	⚫			⚫	⚫	⚫				⚫	General,320x320 only
262	ByteTrack	■■■											⚫	YOLOX/nano,tiny,s,m,l,x,mot17,ablation/128x320,192x320,192x448,192x640,256x320,256x448,256x640,384x640,512x1280,736x1280
264	object_localization_network	■■■											⚫	180x320,240x320,270x480,360x480,360x480,360x640,480x640,720x1280
307	YOLOv7	■■■	⚫	⚫	⚫		⚫	⚫	⚫		⚫		⚫	YOLOv7,YOLOv7-tiny
308	FastestDet	■■■	⚫	⚫	⚫		⚫	⚫	⚫	⚫	⚫		⚫	180x320,256x320,320x480,352x352,352x640,480x640,736x1280
329	YOLOX-PAI	■■■											⚫
332	CrowdDet	■■■											⚫
334	DAMO-YOLO	■■■											⚫
336	PP-YOLOE-Plus	■■■											⚫
337	FreeYOLO	■■■											⚫
341	YOLOv6	■■■	⚫	⚫	⚫		⚫	⚫					⚫
356	EdgeYOLO	■■■											⚫
376	RT-DETR	■■■											⚫	ResNet50,ResNet101,HgNetv2-L,HgNetv2-X
386	naruto_handsign_detection	■■■	⚫	⚫	⚫		⚫						⚫
422	Gold-YOLO-Head-Hand	■■■											⚫	Head,Hand
424	Gold-YOLO-Body	■■■											⚫	Body
425	Gold-YOLO-Body-Head-Hand	■■■											⚫	Body,Head,Hand
426	YOLOX-Body-Head-Hand	■■■	⚫	⚫	⚫								⚫	Body,Head,Hand, tflite float16 XNNPACK boost (ARMv8.2)
434	YOLOX-Body-Head-Hand-Face	■■■											⚫	Body,Head,Hand,Face
441	YOLOX-Body-Head-Hand-Face-Dist	■■■											⚫	Body,Head,Hand,Face,Complex Distorted
442	YOLOX-Body-Head-Face-HandLR-Dist	■■■											⚫	Body,Head,Hands,Left-Hand,Right-Hand,Face,Complex Distorted
444	YOLOX-Foot-Dist	■■■											⚫	Foot,Complex Distorted