Skip to content

Latest commit

 

History

History
43 lines (28 loc) · 1.63 KB

README.md

File metadata and controls

43 lines (28 loc) · 1.63 KB

Nuclio Lighting Flash

This repo builds a base image ghcr.io/greenroom-robotics/nuclio_lighting_flash:latest which is used to conveniently run flash ObjectDetection models in Nuclio / CVAT. It allows you to configure the model head, backbone and checkpoint_path from your nuclio function.yml

Usage

nuctl deploy --project-name cvat \
  --path example/lightningflash-efficientdet-d0 \
  --platform local

Development

Get started

In order to develop you'll want a nuclio instance running on your local machine...

  • docker-compose up to start nuclio.
  • ./scripts/build.sh to build ghcr.io/greenroom-robotics/nuclio_lighting_flash:latest
  • Deploy the example to your nuclio instance:
nuctl deploy --project-name cvat \
  --path example/lightningflash-efficientdet-d0 \
  --platform local

Num Classes

It is important to note that the num_classes (labels) differs for many pretrained model. If you are getting no detections and you think you should, your num_classes is probably wrong. See this for how the coco labels changed over time

Run tests

  • ./scripts/build.sh && docker run ghcr.io/greenroom-robotics/nuclio_lighting_flash:latest to build and run pytests

Release a version

  • Run the Release workflow on github