切换git用户重新进行项目首次归档

FC_ML_Data/FC_ML_Data_Output/Data_Output_File.py

@@ -0,0 +1,12 @@
+import torch
+import pandas as pd
+import json
+
+def tensor_to_excel(tensor,output_path,file_name = "output"):
+    df = pd.DataFrame(tensor.numpy())
+    df.to_excel(output_path + '/' + file_name + '.xlsx', index=False)
+
+def tensor_to_json(tensor,output_path,file_name = "output"):
+    tensor_list = tensor.tolist()  # 转换为Python列表
+    with open(output_path + '/' + file_name +'.json', 'w') as f:
+        json.dump(tensor_list, f)  # 序列化为JSON

FC_ML_Data/FC_ML_Data_Output/Data_Output_Pytorch.py

@@ -0,0 +1,22 @@
+#1 pytorch 自有格式导出
+import torch
+
+def export_model_pt(model,target,name = "model"):
+    script_model = torch.jit.script(model)  # 或 torch.jit.trace(model, input)
+    script_model.save(target + name + ".pt")
+#2 通用格式导出
+def export_model_onnx(model,input_tensor,target,name="model"):
+    torch.onnx.export(model, input_tensor, target+ name + ".onnx")
+#3二进制权重导出
+def export_model_bin(model,target,name = "weights"):
+    torch.save(model.state_dict(), target + name + ".bin")
+
+def export_model(model,target,file_name,name):
+    if name == 'bin':
+        return export_model_bin(model,target,file_name)
+    if name == 'onnx':
+        return export_model_onnx(model,target,file_name)
+    if name == 'pt':
+        return export_model_bin(model,target,file_name)
+    else:
+        raise ValueError(f"不支持的导出类型")

FC_ML_Data/FC_ML_Data_Output/__init__.py

@@ -0,0 +1 @@
+import FC_ML_Data