Skip to content

add, delete, search, change some nodes of .pb or other graph for tensorflow

License

Notifications You must be signed in to change notification settings

fengxiaoshuai/TF_graph_editor

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

OptimizerTooll

About

Optimizer Tooll is a frequently used tool for optimizing neural networks
GraphEdit.py : a tool to modify Graph of tensorflow

install

The appropriate files are downloaded to the local machine

Run


Build node

for example:
import GraphEdit as ge
ge.BuildNdoe().creat_conv_node()


Edit graph

for example:
import GraphEdit as ge
old_graph = './test.pb'
graph_edit = ge.GraphEdit(graph_pb=old_graph, input_node=['input'], output_node=['vgg_16/fc8/squeezed'])
print(graph_edit)

About

add, delete, search, change some nodes of .pb or other graph for tensorflow

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages