Daniel00611 commited on
Commit
a11268f
verified
1 Parent(s): 41c7a5b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +40 -1
README.md CHANGED
@@ -11,4 +11,43 @@ tags:
11
  - biology
12
  datasets:
13
  - WHOI
14
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  - biology
12
  datasets:
13
  - WHOI
14
+ ---
15
+
16
+ # Model Usage
17
+ ```
18
+ import os
19
+ import numpy as np
20
+ from tensorflow.keras.preprocessing import image
21
+ from tensorflow.keras.models import load_model
22
+ from tensorflow.keras.applications.inception_v3 import preprocess_input
23
+ import tensorflow as tf
24
+
25
+
26
+ # Lista de clases
27
+ class_names = ['acanthoica', 'akashiwo', 'alexandrium', 'amoeba', 'amphidinium', 'amylax', 'apedinella', 'asterionellopsis', 'bacillaria', 'bacteriastrum', 'biddulphia', 'calciopappus', 'cerataulina', 'ceratium', 'chaetoceros', 'chrysochromulina', 'cochlodinium', 'corethron', 'corymbellus', 'coscinodiscus', 'cryptophyta', 'cylindrotheca', 'dactyliosolen', 'delphineis', 'dictyocha', 'dinobryon', 'dinophysis', 'ditylum', 'emiliania', 'ephemera', 'eucampia', 'euglena', 'gonyaulax', 'guinardia', 'gyrodinium', 'hemiaulus', 'heterocapsa', 'karenia', 'katodinium', 'kryptoperidinium', 'laboea', 'lauderia', 'leptocylindrus', 'licmophora', 'nanoneis', 'odontella', 'ophiaster', 'ostreopsis', 'oxytoxum', 'paralia', 'parvicorbicula', 'phaeocystis', 'pleuronema', 'pleurosigma', 'polykrikos', 'prorocentrum', 'proterythropsis', 'protoperidinium', 'pseudo-nitzschia', 'pseudochattonella', 'pyramimonas', 'rhabdolithes', 'rhizosolenia', 'scrippsiella', 'skeletonema', 'stephanopyxis', 'syracosphaera', 'thalassionema', 'thalassiosira', 'trichodesmium', 'vicicitus', 'warnowia']
28
+
29
+ def preprocess_image(image_path, target_size=(299, 299)):
30
+ img = image.load_img(image_path, target_size=target_size)
31
+ img_array = image.img_to_array(img)
32
+ img_array = np.expand_dims(img_array, axis=0)
33
+ img_array = preprocess_input(img_array) # Usar la funci贸n de preprocesamiento de InceptionV3
34
+ return img_array
35
+
36
+ # Ruta de la imagen que deseas procesar
37
+ image_path = '/your/image/path'
38
+ img_array = preprocess_image(image_path)
39
+
40
+ # Hacer la predicci贸n
41
+ predictions = model.predict(img_array)[0] # Obtener las probabilidades de la primera (y 煤nica) imagen
42
+
43
+ # Obtener el top 10 de predicciones
44
+ top_10_indices = predictions.argsort()[-10:][::-1] # Ordenar 铆ndices por probabilidad (de mayor a menor)
45
+ top_10_classes = [class_names[i] for i in top_10_indices]
46
+ top_10_probabilities = predictions[top_10_indices]
47
+
48
+ # Mostrar el top 10 de clases con sus probabilidades
49
+ print("Top 10 predicciones:")
50
+ for i in range(10):
51
+ print(f"{top_10_classes[i]}: {top_10_probabilities[i] * 100:.2f}%")
52
+
53
+ ```