Skip to content

Instantly share code, notes, and snippets.

@Dapid

Dapid/retinanet_predict.py

Created November 23, 2023 08:09
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Dapid/44523d82874e7f137fa30df733dd23f4 to your computer and use it in GitHub Desktop.
Save Dapid/44523d82874e7f137fa30df733dd23f4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
retinanet_predict.py
import numpy as np
import keras_cv
model = keras_cv.models.RetinaNet(backbone=keras_cv.models.MobileNetV3Backbone.from_preset('mobilenet_v3_small_imagenet'),
num_classes=2,
bounding_box_format="xywh",
)
image= np.random.randint(0, 255, size=(1, 640, 480, 3), dtype=np.uint8)
model(image)
model.predict(image)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment