Skip to content

BAOfanTing/resnetPTH2ONNX

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
.gitignore
.gitignore
 
 
README.md
README.md
 
 
batch_predict.py
batch_predict.py
 
 
class_indices.json
class_indices.json
 
 
load_weights.py
load_weights.py
 
 
model.py
model.py
 
 
predict.py
predict.py
 
 
pth2onnx.py
pth2onnx.py
 
 
pyh2onnx_test.py
pyh2onnx_test.py
 
 
train.py
train.py
 
 

Repository files navigation

文件结构：

  ├── model.py: ResNet模型搭建
  ├── train.py: 训练脚本
  ├── predict.py: 单张图像预测脚本
  ├── pth2onnx.py: 模型转换脚本
  ├── pth2onnx_test.py: 转换后模型推理脚本
  └── batch_predict.py: 批量图像预测脚本

博客地址

About

将pth模型转换为onnx模型

Resources

Readme
Activity

Stars

0 stars

Watchers

1 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages