Skip to content
  • Jiacong Fang's avatar
    Add TensorFlow and TFLite export (#1127) · 808bcad3
    Jiacong Fang authored
    * Add models/tf.py for TensorFlow and TFLite export
    
    * Set auto=False for int8 calibration
    
    * Update requirements.txt for TensorFlow and TFLite export
    
    * Read anchors directly from PyTorch weights
    
    * Add --tf-nms to append NMS in TensorFlow SavedModel and GraphDef export
    
    * Remove check_anchor_order, check_file, set_logging from import
    
    * Reformat code and optimize imports
    
    * Autodownload model and check cfg
    
    * update --source path, img-size to 320, single output
    
    * Adjust representative_dataset
    
    * Put representative dataset in tfl_int8 block
    
    * detect.py TF inference
    
    * weights to string
    
    * weights to string
    
    * cleanup tf.py
    
    * Add --dynamic-batch-size
    
    * Add xywh normalization to reduce calibration error
    
    * Update requirements.txt
    
    TensorFlow 2.3.1 -> 2.4.0 to avoid int8 quantization error
    
    * Fix imports
    
    Move C3 from models.experimental to models.common
    
    * Add models/tf.py for TensorFlow and TFLite export
    
    * Set auto=F...
    808bcad3
This project manages its dependencies using pip. Learn more