epoch = 21 gives highest accuracy without risk of overfitting.
Install the following packages:
- torch
- torchvision
- PIL
- numpy
- scikit-learn
- tqdm
- json
- onnxruntime
Run the python file to make the onnx
-
Notifications
You must be signed in to change notification settings - Fork 0
SysLords/train-pose-estimation-custom-model
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Custom quantizes models for pose estimation
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published