search

WildMeOrg / wildbook-ia

Wildbook's Image Analysis (WBIA) backend service supporting machine learning for wildlife conservation
https://wildbook.org
Apache License 2.0
77 stars 21 forks source link

Adds multilabel efficientnet #278

Closed LashaO closed 5 months ago

LashaO commented 5 months ago

Adds the new MSv2 multispecies viewpoint labeler. The model works as a mutlilabel classifier under the hood, but the multilabel predictions are translated to single composite labels for compatibility with the current version of WBIA. For example, a prediction of ['up', 'left'] will be translated to 'upleft'. Maximum number of viewpoints that can be joined is 2. This version only supports viewpoint labels, not species.

Configuration for usage:

  "labeler_algo": "efficientnet",
  "labeler_model_tag": "msv2_multilabel_effnet_v0",
  "use_labeler_species": False,