‘export_formats‘ from ‘export‘

残影飞雪

于 2024-07-29 09:06:50 发布

阅读量201

点赞数 4

分类专栏： YOLO Python 文章标签： neo4j

本文链接：https://blog.csdn.net/canyingfeixue/article/details/140762028

版权

Python 同时被 2 个专栏收录

51 篇文章 0 订阅

订阅专栏

YOLO

13 篇文章 0 订阅

订阅专栏

修改 model/common.py 616行

def export_formats(): 

            # YOLOv5 export formats 

            x = [ 

                ['PyTorch', '-', '.pt', True, True], 

                ['TorchScript', 'torchscript', '.torchscript', True, True], 

                ['ONNX', 'onnx', '.onnx', True, True], 

                ['OpenVINO', 'openvino', '_openvino_model', True, False], 

                ['TensorRT', 'engine', '.engine', False, True], 

                ['CoreML', 'coreml', '.mlmodel', True, False], 

                ['TensorFlow SavedModel', 'saved_model', '_saved_model', True, True], 

                ['TensorFlow GraphDef', 'pb', '.pb', True, True], 

                ['TensorFlow Lite', 'tflite', '.tflite', True, False], 

                ['TensorFlow Edge TPU', 'edgetpu', '_edgetpu.tflite', False, False], 

                ['TensorFlow.js', 'tfjs', '_web_model', False, False], 

                ['PaddlePaddle', 'paddle', '_paddle_model', True, True],] 

            return pd.DataFrame(x, columns=['Format', 'Argument', 'Suffix', 'CPU', 'GPU']) 

        # Return model type from model path, i.e. path='path/to/model.onnx' -> type=onnx 
        # types = [pt, jit, onnx, xml, engine, coreml, saved_model, pb, tflite, edgetpu, tfjs, paddle]