MCPcopy Index your code
hub / github.com/PINTO0309/PINTO_model_zoo

github.com/PINTO0309/PINTO_model_zoo @1.0.0 sqlite

repository ↗ · DeepWiki ↗ · release 1.0.0 ↗
4,877 symbols 17,100 edges 1,061 files 829 documented · 17%
README

PINTO_model_zoo

CodeQL DOI Ask DeepWiki

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

image

image

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
### 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.

Core symbols most depended-on inside this repo

load
called by 826
018_EfficientDet/01_float32/inference.py
get
called by 624
407_Generalizing_Gaze_Estimation/demo/demo_gaze_estimation.py
get_tf_edges_from
called by 490
148_LapDepth/openvino2tensorflow_custom.py
rectangle
called by 277
461_YOLOv9-Phone/yolov9/utils/plots.py
export
called by 276
018_EfficientDet/01_float32/inference.py
info
called by 251
459_YOLOv9-Wholebody25/yolov9/models/yolo.py
clip
called by 177
041_DBFace/common.py
extrapolation_of_layers
called by 162
148_LapDepth/openvino2tensorflow_custom.py

Shape

Function 2,150
Method 2,016
Class 711

Languages

Python100%

Modules by API surface

468_YOLOv9-Wholebody28-Refine/yolov9/models/yolo.py56 symbols
464_YOLOv9-Wholebody28/yolov9/models/yolo.py56 symbols
463_YOLOv9-Shoulder-Elbow-Knee/yolov9/models/yolo.py56 symbols
461_YOLOv9-Phone/yolov9/models/yolo.py56 symbols
459_YOLOv9-Wholebody25/yolov9/models/yolo.py56 symbols
468_YOLOv9-Wholebody28-Refine/yolov9/models/quantize.py50 symbols
464_YOLOv9-Wholebody28/yolov9/models/quantize.py50 symbols
463_YOLOv9-Shoulder-Elbow-Knee/yolov9/models/quantize.py50 symbols
461_YOLOv9-Phone/yolov9/models/quantize.py50 symbols
459_YOLOv9-Wholebody25/yolov9/models/quantize.py50 symbols
458_YOLOv9-Discrete-HeadPose-Yaw/yolov9/models/quantize.py50 symbols
457_YOLOv9-Wholebody17/yolov9/models/quantize.py50 symbols

For agents

$ claude mcp add PINTO_model_zoo \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact