Skip to content

Commit fbabcad

Browse files
committed
Remove redunt input layer
1 parent a9c3f9a commit fbabcad

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

keras2ncnn/graph_optimizer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,10 @@ def removing_unused_nodes(graph):
77

88
for target_node_name in graph.get_graph().keys():
99
if graph.get_node_attr(target_node_name)[
10-
'layer']['class_name'] in UNUSED_NODES:
10+
'layer']['class_name'] in UNUSED_NODES or \
11+
( graph.get_node_attr(target_node_name)[
12+
'layer']['class_name'] == 'InputLayer' and \
13+
len(graph.get_node_inbounds(target_node_name)) != 0 ):
1114
for layer_name in graph.get_graph().keys():
1215
if target_node_name in graph.get_graph()[
1316
layer_name]['inbounds']:

0 commit comments

Comments
 (0)