restyled

MarsTechHAN · Jan 22, 2021 · 9841e26 · 9841e26
1 parent 68dd7d6
commit 9841e26
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/keras2ncnn/keras_converter.py b/keras2ncnn/keras_converter.py
@@ -986,6 +986,7 @@ def parse_keras_graph(
                 ncnn_graph_helper.set_node_attr(
                     graph_head + '_input', {
                         'type': 'Input', 'param': ncnn_graph_attr, 'binary': []})
-                ncnn_graph_helper.add_node_inbounds(graph_head, graph_head + '_input')
-
-        ncnn_graph_helper.refresh()
+                ncnn_graph_helper.add_node_inbounds(
+                    graph_head, graph_head + '_input')
+
+        ncnn_graph_helper.refresh()
diff --git a/keras2ncnn/keras_debugger.py b/keras2ncnn/keras_debugger.py
@@ -6,6 +6,7 @@
 import shutil
 import sys
 
+
 class KerasDebugger:
     ncnn_prog_template = \
         '''
@@ -99,8 +100,8 @@ def init_env(self):
                 'cmake'
             ]
 
-            subprocess.run(
-                ['python3', '-m', 'pip', 'install', '--upgrade', 'virtualenv>=15.0.0'])
+            subprocess.run(['python3', '-m', 'pip', 'install',
+                            '--upgrade', 'virtualenv>=15.0.0'])
 
             for util in required_utils:
                 res = sp.find_executable(util)
@@ -111,7 +112,7 @@ def init_env(self):
                     return -1
 
             # Setup virtualenv
-            import virtualenv # pylint: disable=import-outside-toplevel
+            import virtualenv  # pylint: disable=import-outside-toplevel
             virtualenv.create_environment(self.tmp_dir)
 
             activator_filename = os.path.join(