FishNet: AI For Fish Stock Estimation
The attached model was trained on a 63000 dataset of fish images belonging to 163 species. First, we trained a detectron2 model to detect and segment fish and fiduciary markers on a board. The detectron2 model was written in PyTorch, and the final classifier is a ResNet50 keras model.
Below is an example of how to use the 2 models.
Packages
Load Models
load an image and transform
run the segmentation model
visualize
classify the fish
Inference Providers
NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API:
The model has no library tag.