Spaces:
Sleeping
Sleeping
Commit
•
21a60f0
1
Parent(s):
2637de5
Init app
Browse files- .gitattributes +2 -0
- app.py +44 -0
- daurade.jpg +3 -0
- model.pth +3 -0
- requirements.txt +5 -0
.gitattributes
CHANGED
@@ -32,3 +32,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
32 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
33 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
34 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
32 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
33 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
34 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
35 |
+
model.pth filter=lfs diff=lfs merge=lfs -text
|
36 |
+
daurade.jpg filter=lfs diff=lfs merge=lfs -text
|
app.py
ADDED
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
import gradio as gr
|
3 |
+
from PIL import Image
|
4 |
+
from torchvision import transforms
|
5 |
+
import timm
|
6 |
+
|
7 |
+
classes = ['la_badèche', 'la_blennie_coiffée', 'la_blennie_de_caneva', 'la_blennie_de_roux', 'la_blennie_de_zvonimir', 'la_blennie_diabolo', 'la_blennie_gattorugine', 'la_blennie_palmicorne', 'la_blennie_paon', 'la_blennie_pilicorne', 'la_blennie_sphynx', 'la_bogue', 'la_bécune_bouche_jaune', 'la_canthare', 'la_carangue_dentue', 'la_castagnole', 'la_coquette', 'la_dorade_royale', 'la_girelle', 'la_girelle-paon', 'la_gonnelle', 'la_grande_seriole', 'la_grande_vive', 'la_morue', 'la_mostelle_de_roche', 'la_murène_commune', 'la_palomine', 'la_petite_rascasse_rouge', 'la_rascasse_brune', 'la_rascasse_de_madère', 'la_saupe', 'la_sole_commune', 'la_vielle', 'le_baliste_commun', 'le_bar_commun', 'le_barbier_commun', 'le_chabot_buffle', 'le_chabot_commun', 'le_chapon', 'le_congre', 'le_corb', 'le_crénilabre_cendré', 'le_crénilabre_de_melops', 'le_crénilabre_méditerranéen', 'le_crénilabre_ocellé', 'le_crénilabre_tanche', 'le_crénilabre_à_5_taches', 'le_cténolabre', 'le_denté_commun', 'le_dragonnet_lyre', 'le_flet', 'le_gobie_de_sarato', 'le_gobie_marbré', 'le_gobie_nageur', 'le_gobie_noir', 'le_gobie_paganel', 'le_gobie_svelte', 'le_gobie_tacheté', 'le_gobie_varié', 'le_gobie_à_bouche_rouge', 'le_gobie_à_grosse_tête', 'le_gobie_à_tête_jaune', 'le_grondin-perlon', 'le_grondin_camard', 'le_joel', 'le_marbré', 'le_merle', 'le_mordocet', 'le_mulet_labeon', 'le_mulet_lippu', 'le_mérou_brun', 'le_nérophis_lombricoïde', 'le_poisson-lézard_rayé', 'le_poisson-perroquet', 'le_poisson_lapin_à_queue_tronquée', 'le_poisson_lapin_à_ventre_strié', 'le_rason', 'le_rombou', 'le_rouget-barbet_de_roche', 'le_rouget-barbet_de_vase', 'le_sar_commun', 'le_sar_à_grosses_lèvres', 'le_sar_à_museau_pointu', 'le_sar_à_tête_noire', 'le_serran_-chevrette', 'le_serran_ecriture', 'le_siphonostome_atlantique', 'le_sparaillon', 'le_sublet', 'le_syngnathe_aiguille', 'le_tripterygion_rouge', 'le_triptérygion_jaune', 'le_tryptérigion_nain', 'l’anguille', 'l’apogon_commun', 'l’hippocampe_moucheté', 'l’hippocampe_à_nez_court', 'l’hippocampe_à_ramules', 'l’oblade', 'l’orphie', 'l’épinoche']
|
8 |
+
|
9 |
+
# Load the model
|
10 |
+
model_path = "model.pth"
|
11 |
+
model = timm.create_model('resnet34', pretrained=False, num_classes=101)
|
12 |
+
model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
|
13 |
+
model = model.to('cpu')
|
14 |
+
model.eval()
|
15 |
+
|
16 |
+
|
17 |
+
# Preprocessing function
|
18 |
+
def preprocess(image):
|
19 |
+
_preprocess = transforms.Compose([
|
20 |
+
transforms.Resize((224, 224)),
|
21 |
+
transforms.ToTensor(),
|
22 |
+
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
|
23 |
+
])
|
24 |
+
return _preprocess(image)
|
25 |
+
|
26 |
+
|
27 |
+
# Inference function
|
28 |
+
def fish_classifier(img_input):
|
29 |
+
img = Image.fromarray(img_input.astype('uint8'), 'RGB')
|
30 |
+
input_tensor = preprocess(img).unsqueeze(0)
|
31 |
+
|
32 |
+
with torch.no_grad():
|
33 |
+
pred = model(input_tensor)[0]
|
34 |
+
pred = torch.nn.functional.softmax(pred, dim=0)
|
35 |
+
return {classes[i]: float(pred[i]) for i in range(len(classes))}
|
36 |
+
|
37 |
+
|
38 |
+
# Gradio interface
|
39 |
+
gr.Interface(
|
40 |
+
fn=fish_classifier,
|
41 |
+
inputs=gr.Image(shape=(224, 224)),
|
42 |
+
outputs=gr.Label(num_top_classes=5),
|
43 |
+
examples=["dorade.jpg"]
|
44 |
+
).launch()
|
daurade.jpg
ADDED
![]() |
Git LFS Details
|
model.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:64f0bdb97c037efa5f8157fe36f83a761d7cbff6cb3cd0bb52209ba8a35654e0
|
3 |
+
size 85488161
|
requirements.txt
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
gradio
|
2 |
+
Pillow
|
3 |
+
torch
|
4 |
+
torchvision
|
5 |
+
timm
|