Skip to content

Add model for Kenya & Uganda moths from UKCEH / Turing #78

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
mihow merged 9 commits into from
Apr 15, 2025
Merged

Add model for Kenya & Uganda moths from UKCEH / Turing #78

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
f4e8316
Add UKCEH models
KatrionaGoldmann Jan 27, 2025
5645129
Add UKCEH models
KatrionaGoldmann Jan 27, 2025
3e6d5c0
Merge branch 'RolnickLab:main' into main
KatrionaGoldmann Jan 27, 2025
786e6ce
uncomment models
KatrionaGoldmann Jan 27, 2025
b3b5364
Merge origin
KatrionaGoldmann Jan 27, 2025
e064385
fix catch
KatrionaGoldmann Jan 27, 2025
73f1e7e
wrong labels file
KatrionaGoldmann Jan 28, 2025
ee234ae
Merge branch 'main' of https://github.com/AMI-system/ami-data-compani…
mihow Apr 15, 2025
79ba87d
Only add models that have weights available (Kenya)
mihow Apr 15, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions trapdata/api/api.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
MothClassifierQuebecVermont,
MothClassifierTuringAnguilla,
MothClassifierTuringCostaRica,
MothClassifierTuringKenyaUganda,
MothClassifierUKDenmark,
)
from .models.localization import APIMothDetector
Expand All @@ -46,6 +47,7 @@
"uk_denmark_moths_2023": MothClassifierUKDenmark,
"costa_rica_moths_turing_2024": MothClassifierTuringCostaRica,
"anguilla_moths_turing_2024": MothClassifierTuringAnguilla,
"kenya-uganda_moths_turing_2024": MothClassifierTuringKenyaUganda,
"global_moths_2024": MothClassifierGlobal,
"moth_binary": MothClassifierBinary,
"insect_orders_2025": InsectOrderClassifier,
Expand Down
7 changes: 7 additions & 0 deletions trapdata/api/models/classification.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
QuebecVermontMothSpeciesClassifier2024,
TuringAnguillaSpeciesClassifier,
TuringCostaRicaSpeciesClassifier,
TuringKenyaUgandaSpeciesClassifier,
UKDenmarkMothSpeciesClassifier2024,
)

Expand Down Expand Up @@ -187,6 +188,12 @@ class MothClassifierTuringAnguilla(APIMothClassifier, TuringAnguillaSpeciesClass
pass


class MothClassifierTuringKenyaUganda(
APIMothClassifier, TuringKenyaUgandaSpeciesClassifier
):
pass


class MothClassifierGlobal(APIMothClassifier, GlobalMothSpeciesClassifier):
pass

Expand Down
23 changes: 21 additions & 2 deletions trapdata/ml/models/classification.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -406,8 +406,27 @@ class TuringCostaRicaSpeciesClassifier(SpeciesClassifier, Resnet50Classifier_Tur
class TuringAnguillaSpeciesClassifier(SpeciesClassifier, Resnet50Classifier_Turing):
name = "Turing Anguilla Species Classifier"
description = "Trained on 28th June 2024 by Turing team using Resnet50 model."
weights_path = "https://object-arbutus.cloud.computecanada.ca/ami-models/moths/classification/turing-anguilla_v01_resnet50_2024-06-28-17-01_state.pt"
labels_path = "https://object-arbutus.cloud.computecanada.ca/ami-models/moths/classification/01_anguilla_data_category_map.json"
weights_path = (
"https://object-arbutus.cloud.computecanada.ca/ami-models/moths/classification/"
"turing-anguilla_v01_resnet50_2024-06-28-17-01_state.pt"
)
labels_path = (
"https://object-arbutus.cloud.computecanada.ca/ami-models/moths/classification/"
"01_anguilla_data_category_map.json"
)


class TuringKenyaUgandaSpeciesClassifier(SpeciesClassifier, Resnet50Classifier_Turing):
name = "Turing Kenya and Uganda Species Classifier"
description = "Trained on 19th November 2024 by Turing team using Resnet50 model."
weights_path = (
"https://object-arbutus.cloud.computecanada.ca/ami-models/moths/classification/"
"turing-kenya-uganda_v01_resnet50_2024-11-19-18-44_state.pt"
)
labels_path = (
"https://object-arbutus.cloud.computecanada.ca/ami-models/moths/classification/"
"01_kenya-uganda_data_category_map.json"
)


class TuringUKSpeciesClassifier(SpeciesClassifier, Resnet50Classifier_Turing):
Expand Down