• Glenn Jocher's avatar
    Add TensorFlow formats to `export.py` (#4479) · c3a93d78
    Glenn Jocher 提交于
    * Initial commit
    
    * Remove unused export_torchscript return
    
    * ROOT variable
    
    * Add prefix to fcn arg
    
    * fix ROOT
    
    * check_yaml into run()
    
    * interim fixes
    
    * imgsz=(320, 320)
    
    * Hardcode tf_raw_resize False
    
    * Finish opt elimination
    
    * Update representative_dataset_gen()
    
    * Update export.py with TF methods
    
    * SiLU and GraphDef fixes
    
    * file_size() directory handling feature
    
    * export fixes
    
    * add lambda: to representative_dataset
    
    * Detect training False default
    
    * Fuse false for TF models
    
    * Embed agnostic NMS arguments
    
    * Remove lambda
    
    * TensorFlow.js export success
    
    * Add pb to Usage
    
    * Add *_tfjs_model/ to ignore files
    
    * prepend YOLOv5 to function headers
    
    * Remove end --- comments
    
    * parameterize tfjs export pb file
    
    * update run() data default /ROOT
    
    * update --include help
    
    * update imports
    
    * return ct_model
    
    * Consolidate TFLite export
    
    * pb prerequisite to tfjs
    
    * TF modules CamelCase
    
    * Remove exports from tf.py and cleanup
    
    * pass agnostic NMS arguments
    
    * CI
    
    * CI
    
    * ignore *_web_model/
    
    * Add tensorflow to CI dependencies
    
    * CI tensorflow-cpu
    
    * Update requirements.txt
    
    * Remove tensorflow check_requirement
    
    * CI coreml tfjs
    
    * export only onnx torchscript
    
    * reorder exports torchscript first
    c3a93d78
.dockerignore 3.6 KB