Julia-Amadio
commited on
Commit
·
0b34608
1
Parent(s):
667a199
Últimos ajustes.
Browse files- README.md +2 -2
- feature_extractor_single.py +1 -1
- requirements.txt +4 -12
README.md
CHANGED
|
@@ -1,13 +1,13 @@
|
|
| 1 |
---
|
| 2 |
title: Soja Api
|
| 3 |
-
emoji:
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: blue
|
| 6 |
sdk: docker
|
| 7 |
app_port: 7860
|
| 8 |
pinned: false
|
| 9 |
license: mit
|
| 10 |
-
short_description: API for
|
| 11 |
---
|
| 12 |
|
| 13 |
## Descrição
|
|
|
|
| 1 |
---
|
| 2 |
title: Soja Api
|
| 3 |
+
emoji: 🌱
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: blue
|
| 6 |
sdk: docker
|
| 7 |
app_port: 7860
|
| 8 |
pinned: false
|
| 9 |
license: mit
|
| 10 |
+
short_description: test API for feature extraction.
|
| 11 |
---
|
| 12 |
|
| 13 |
## Descrição
|
feature_extractor_single.py
CHANGED
|
@@ -39,7 +39,7 @@ def process_single_image(image_path: str, output_dir: str = "processed"):
|
|
| 39 |
"""
|
| 40 |
#1) Extração de features
|
| 41 |
extractor = FeatureExtractor()
|
| 42 |
-
#
|
| 43 |
features = extractor.extract_convnext(image_path)
|
| 44 |
|
| 45 |
#NÃO PRECISAMOS MAIS SALVAR O ARQUIVO .NPY
|
|
|
|
| 39 |
"""
|
| 40 |
#1) Extração de features
|
| 41 |
extractor = FeatureExtractor()
|
| 42 |
+
#Passa o CAMINHO ORIGINAL da imagem (ex: /tmp/temp_foto.jpg)
|
| 43 |
features = extractor.extract_convnext(image_path)
|
| 44 |
|
| 45 |
#NÃO PRECISAMOS MAIS SALVAR O ARQUIVO .NPY
|
requirements.txt
CHANGED
|
@@ -1,26 +1,18 @@
|
|
| 1 |
-
#
|
| 2 |
-
# Núcleo da API
|
| 3 |
-
# =====================================
|
| 4 |
fastapi==0.115.0
|
| 5 |
uvicorn[standard]==0.30.6
|
| 6 |
python-multipart==0.0.9
|
| 7 |
|
| 8 |
-
#
|
| 9 |
-
# Deep Learning / Modelos (CPU)
|
| 10 |
-
# =====================================
|
| 11 |
torch==2.2.2+cpu
|
| 12 |
torchvision==0.17.2+cpu
|
| 13 |
-f https://download.pytorch.org/whl/torch_stable.html
|
| 14 |
|
| 15 |
transformers==4.39.3
|
| 16 |
|
| 17 |
-
#
|
| 18 |
-
# Processamento de imagem
|
| 19 |
-
# =====================================
|
| 20 |
pillow==10.3.0
|
| 21 |
numpy==1.26.4
|
| 22 |
|
| 23 |
-
#
|
| 24 |
-
# Utilitários gerais
|
| 25 |
-
# =====================================
|
| 26 |
requests==2.32.3
|
|
|
|
| 1 |
+
#Núcleo da API
|
|
|
|
|
|
|
| 2 |
fastapi==0.115.0
|
| 3 |
uvicorn[standard]==0.30.6
|
| 4 |
python-multipart==0.0.9
|
| 5 |
|
| 6 |
+
#Deep Learning / Modelos (CPU)
|
|
|
|
|
|
|
| 7 |
torch==2.2.2+cpu
|
| 8 |
torchvision==0.17.2+cpu
|
| 9 |
-f https://download.pytorch.org/whl/torch_stable.html
|
| 10 |
|
| 11 |
transformers==4.39.3
|
| 12 |
|
| 13 |
+
#Processamento de imagem
|
|
|
|
|
|
|
| 14 |
pillow==10.3.0
|
| 15 |
numpy==1.26.4
|
| 16 |
|
| 17 |
+
#Utilitários gerais
|
|
|
|
|
|
|
| 18 |
requests==2.32.3
|