keras-nlp icon indicating copy to clipboard operation
keras-nlp copied to clipboard

YOLOV8 port to keras-hub

Open oarriaga opened this issue 1 year ago • 14 comments
trafficstars

This PR ports YOLOV8 from keras-cv to keras-hub (https://github.com/keras-team/keras-hub/issues/176). All necessary YOLOV8 functions are now found inside keras-hub:

  • Add CIOU loss.
  • Add missing masking functionality in the bounding_boxes module.
  • Add multibackend non maximum supression layer.
  • Add label encoder.
  • Build basic abstract object detector task class.
  • Add YOLOV8 backbone and detector.

Missing steps include:

  • Upload previous presets to Kaggle.
  • Remove skipping tests with presets.
  • Add colab with basic functionality.
  • Add weight transfer script from keras-cv to keras-hub.
  • Add training script.

oarriaga avatar Oct 01 '24 18:10 oarriaga