Skip to content

Peeta586/pytorch1.0_to_caffe

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pytorch1.0_to_caffe

Converting pytorch1.0 to caffe autmatically

current status

only finish converting pytorch v1.0 to caffe prototxt, and testing on mobilenetv2, shufflenetv2 and biseNet successfully. but some layers in pytorch with no corresponding layers in caffe are directly converted into empty layers that need to be modified manually.

add main.py

this script can convert pytorch model weights into caffemodel

About

Converting pytorch1.0 to caffe autmatically

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages