How to retrain simple MLP (palm_detection_full_inf_post_192x192.onnx) model with custom hand dataset ?

PINTO0309 / PINTO_model_zoo

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.

MIT License

3.43k stars 560 forks source link

Issue Type

Bug

OS

Ubuntu

OS architecture

x86_64

Programming Language

Python

Framework

PyTorch, ONNX, TensorRT

Model name and Weights/Checkpoints URL

Model Name : palm_detection_full_inf_post_192x192.onnx

URL : https://github.com/PINTO0309/hand-gesture-recognition-using-onnx/tree/main/model/palm_detection

Description

How to retrain a simple MLP (palm_detection_full_inf_post_192x192.onnx) model with a custom hand dataset? Do you suggest any guidance or procedure to follow?

Relevant Log Output

Thank you Sir

URL or source code for simple inference testing code

https://github.com/PINTO0309/hand-gesture-recognition-using-onnx/tree/main/model/palm_detection