fix bug

src/Drive_Converter.py

@@ -131,7 +131,7 @@ def get_path_from_db(modelid,seetanet_model):
  
  
 	     #Driver converter model by  params and model graph 
 	     #Driver converter model by  params and model graph 
 	     result = Run_Converter(params_name,graph_name,seetanet_model)
 	     result = Run_Converter(params_name,graph_name,seetanet_model)
-             if result is "true":
+             if result != True:
                   return None,None 
                   return None,None 
             
             
 	     #Run_Converter("model-0015.params","model-symbol.json",seetanet_model)
 	     #Run_Converter("model-0015.params","model-symbol.json",seetanet_model)

src/GProgress.py

@@ -32,5 +32,8 @@ def set_err_msg_var(err_msg):
 
 
 def Post_return():
 def Post_return():
    post_return = { "posttype": GProgress_Var.posttype, "progress" : GProgress_Var.progress, "pool_id": GProgress_Var.pool_id, "err_msg":GProgress_Var.err_msg }
    post_return = { "posttype": GProgress_Var.posttype, "progress" : GProgress_Var.progress, "pool_id": GProgress_Var.pool_id, "err_msg":GProgress_Var.err_msg }
+   print(post_return)
+   try:
      requests.post(post_url, data=post_return)
      requests.post(post_url, data=post_return)
-   
+   except requests.RequestException as e:
+     print(e)

src/config.py

-DATA_NAME = 'data'
+DATA_NAME = 'data_0'
 OUTPUT_LAYER = ''
 OUTPUT_LAYER = ''
 #MODEL_PARAM = '/home/dev01/workshop/projects/MXNet2SeetaNet/model-0015.params'
 #MODEL_PARAM = '/home/dev01/workshop/projects/MXNet2SeetaNet/model-0015.params'
 #MODEL_JSON = '/home/dev01/workshop/projects/MXNet2SeetaNet/model-symbol.json'
 #MODEL_JSON = '/home/dev01/workshop/projects/MXNet2SeetaNet/model-symbol.json'

src/converter.py

@@ -431,7 +431,7 @@ def Run_Converter(model_param,model_json,seetanet_model):
       logging.info('start to convert model parameters')
       logging.info('start to convert model parameters')
       converter.convert(seetanet_model)
       converter.convert(seetanet_model)
       logging.info('convert success!!!')
       logging.info('convert success!!!')
-      return true
+      return True
    except Exception, e:
    except Exception, e:
       return None
       return None
       #print(e)
       #print(e)

src/mxnet_graph.py

@@ -214,6 +214,7 @@ def construct_graph(json_file):
 
 
 
 
 def load_graph(json_file):
 def load_graph(json_file):
+   try:
     logging.info('construct graph')
     logging.info('construct graph')
     graph = construct_graph(json_file)
     graph = construct_graph(json_file)
     logging.info('remove Flatten layer')
     logging.info('remove Flatten layer')
@@ -226,6 +227,9 @@ def load_graph(json_file):
     set_idx(graph)
     set_idx(graph)
     logging.info('load graph over')
     logging.info('load graph over')
     return graph
     return graph
+   except Exception, e:
+    print(e)
+    return None
 
 
 
 
 # if __name__ == '__main__':
 # if __name__ == '__main__':