YOLOV8 not working with JAX Backend

keras-team / keras-cv

Industry-strength Computer Vision workflows with Keras

Other

1.01k stars 330 forks source link

YOLOV8 not working with JAX Backend #2182

Closed IMvision12 closed 10 months ago

IMvision12 commented 11 months ago

I am not able to import the pre-trained backbone yolov8 model using from_preset.

colab : https://colab.research.google.com/drive/1aXzULmIP-MGJ7nxjBuueqR7RUcDvkAvf?usp=sharing

@ianstenbit

divyashreepathihalli commented 10 months ago

Hi @IMvision12! This error should be resolved now. Please remove the keras-core install. Use the following setup code for your colab(this will be simplified soon)

!pip install --upgrade git+https://github.com/keras-team/keras-cv -q
!pip install --upgrade torch
!pip install --upgrade tensorflow
!pip install --upgrade jax
!pip install --upgrade keras

I just tested your colab and it seems to be working now.