Update README.md
Browse files
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 |
+
```
|