Machine Learning Notes - [Machine Learning Case] Custom multi-head + multi-label prediction based on KerasCV pre-training model

1. KerasCV

        KerasCV is a library of modular computer vision components that work natively with TensorFlow, JAX, or PyTorch. Built on top of Keras Core , these models, layers, metrics, callbacks, etc. can be trained and serialized in any framework, and reused in another without costly migrations.

        KerasCV can be understood as a horizontal extension of the Keras API: components are new first-party Keras objects that are too specialized to be added to core Keras. They get the same level of refinement and backward compatibility guarantees as the core Keras API, and are maintained by the Keras team.

        My KerasCV API assists in performing common computer vision tasks such as data augmentation, classification, object detection, segmentation, image generation, and more. Applied computer vision engineers can leverage KerasCV to quickly assemble production-grade, state-of-the-art training and inference pipelines for all of these common tasks.

1. List of supported models

Preset name Model Parameters Description
csp_darknet_tiny_imagenet

Guess you like

Origin blog.csdn.net/bashendixie5/article/details/132521623