app update

app.py

@@ -4,11 +4,6 @@
 # 功能描述：多选，多项模型转换和打包下载
 
 import os
-
-os.system("pip install pip -U")
-os.system("pip install nvidia-pyindex")
-os.system("pip install nvidia-tensorrt")
-
 import shutil
 import time
 import zipfile
@@ -46,18 +41,14 @@ def zipDir(origin_dir, compress_file):
     zip.close()
 
 
-# params_include_list = ["torchscript", "onnx", "openvino", "engine", "coreml", "saved_model", "pb", "tflite", "tfjs"]
+# params_include_list = ["torchscript", "onnx", "openvino", "coreml", "saved_model", "pb", "tflite", "tfjs"]
 def cb_opt(weight_name, btn_model_list, params_include_list):
 
     for i in range(len(btn_model_list)):
         if btn_model_list[i]:
             st.info(f"正在转换{params_include_list[i]}......")
             s = time.time()
-            if i == 3:
-                os.system(
-                    f'python export.py --weights ./weights/{weight_name} --include {params_include_list[i]} --device 0')
-            else:
-                os.system(f'python export.py --weights ./weights/{weight_name} --include {params_include_list[i]}')
+            os.system(f'python export.py --weights ./weights/{weight_name} --include {params_include_list[i]}')
             e = time.time()
             st.success(f"{params_include_list[i]}转换完成，用时{round((e-s), 2)}秒")
 
@@ -93,7 +84,7 @@ def main():
             cb_torchscript = st.checkbox('TorchScript')
             cb_onnx = st.checkbox('ONNX')
             cb_openvino = st.checkbox('OpenVINO')
-            cb_engine = st.checkbox('TensorRT')
+            # cb_engine = st.checkbox('TensorRT')
             cb_coreml = st.checkbox('CoreML')
             cb_saved_model = st.checkbox('TensorFlow SavedModel')
             cb_pb = st.checkbox('TensorFlow GraphDef')
@@ -104,7 +95,7 @@ def main():
             btn_convert = st.button('转换')
 
             btn_model_list = [
-                cb_torchscript, cb_onnx, cb_openvino, cb_engine, cb_coreml, cb_saved_model, cb_pb, cb_tflite, cb_tfjs]
+                cb_torchscript, cb_onnx, cb_openvino, cb_coreml, cb_saved_model, cb_pb, cb_tflite, cb_tfjs]
 
             params_include_list = [
                 "torchscript", "onnx", "openvino", "engine", "coreml", "saved_model", "pb", "tflite", "tfjs"]

requirements.txt

@@ -32,11 +32,9 @@ seaborn
 coremltools  # CoreML export
 onnx  # ONNX export
 onnx-simplifier  # ONNX simplifier
-onnxruntime-gpu
-# nvidia-pyindex==1.0.9
-# nvidia-tensorrt==8.4.1.5
+onnxruntime
 scikit-learn==0.19.2 # CoreML quantization
-tensorflow  # TFLite export
+tensorflow-cpu  # TFLite export
 tensorflowjs  # TF.js export
 openvino-dev  # OpenVINO export