Spaces:
Running
Running
GitHub Actions commited on
Commit ·
0241c3f
1
Parent(s): 5d925b4
Sync from GitHub Actions
Browse files- app/inference.py +1 -1
- autocatalog/data/preprocessing.py +1 -1
app/inference.py
CHANGED
|
@@ -58,4 +58,4 @@ def render_metrics(metrics):
|
|
| 58 |
col1.metric("Average Accuracy", format_percent(overall.get("average_accuracy", 0)))
|
| 59 |
col2.metric("Weighted F1", format_percent(overall.get("average_weighted_f1", 0)))
|
| 60 |
col3.metric("Top-3 Accuracy", format_percent(overall.get("average_top3_accuracy", 0)))
|
| 61 |
-
col4.metric("Test Samples", f"{overall.get('
|
|
|
|
| 58 |
col1.metric("Average Accuracy", format_percent(overall.get("average_accuracy", 0)))
|
| 59 |
col2.metric("Weighted F1", format_percent(overall.get("average_weighted_f1", 0)))
|
| 60 |
col3.metric("Top-3 Accuracy", format_percent(overall.get("average_top3_accuracy", 0)))
|
| 61 |
+
col4.metric("Test Samples", f"{overall.get('samples', 0):,}")
|
autocatalog/data/preprocessing.py
CHANGED
|
@@ -5,7 +5,7 @@ COLOR_IMAGE_SIZE = 128
|
|
| 5 |
COLOR_FEATURE_DIM = 37
|
| 6 |
|
| 7 |
def extract_color_features(image, image_size=COLOR_IMAGE_SIZE):
|
| 8 |
-
image = image.
|
| 9 |
margin = int(image_size * 0.10)
|
| 10 |
|
| 11 |
image = image.crop((
|
|
|
|
| 5 |
COLOR_FEATURE_DIM = 37
|
| 6 |
|
| 7 |
def extract_color_features(image, image_size=COLOR_IMAGE_SIZE):
|
| 8 |
+
image = image.convert("RGB").resize((image_size, image_size))
|
| 9 |
margin = int(image_size * 0.10)
|
| 10 |
|
| 11 |
image = image.crop((
|