Skip to content

0.7 Migrated to tf.keras API and TensorFlow 2.3
Compare
Choose a tag to compare
@AIWintermuteAI AIWintermuteAI released this 30 Dec 15:10
· 73 commits to master since this release
0.7
5f18092
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

Major changes:

  • All scripts are migrated to use tf.keras and TensorFlow 2.3. Works with tf 2.4 as well, but mostly tested with 2.3 at this point. All training scripts are working properly, K210/tflite conversion scripts are tested and fully-functioning, Edge TPU conversion partially working. Due to tensorflow changing default converter to MLIR (Multi-Level Intermediate Representation) converter and multiple breaking changes in API, onnx/openVINO converters are not functioning yet, but will be fixed by next release.
  • Augmentation pipeline is improved, now default augmentation pipeline is "softer" - previously too many augmentations were applied simultaneously, which might have hindered model training and lead to under-fitting.
  • Warm-up learning rate scheduler has been added and made default lerning rate control callback, as opposed to Reduce Learning rate on Plateau from previous versions. Read more about it in https://arxiv.org/abs/1812.01187v2 . Preliminary testing on people detection task, shows significant increase in validation mAP.

Minor changes:

  • Plots of accuracy/loss/mAP are depreciated and replaced with Tensorboard logs. Colab supports interactive Tensorboard interface for tf >2.0, so overall it is better choice for training monitoring/analysis.
  • Added scripts for Edge TPU (this section to be expanded)
  • TODO tasks moved to Github projects.
Assets 2
Loading