Image Classification
Keras
LiteRT
English
Research
Education
Science and Technology
Artificial Intelligence
Computer Science
Computer Vision
CNN
Image
Keras
TensorFlow
Python
Instructions to use me-aas/EinsteinNet with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use me-aas/EinsteinNet with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://me-aas/EinsteinNet") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -68,14 +68,14 @@ This model is not intended for general-purpose natural image classification (lik
|
|
| 68 |
---
|
| 69 |
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
| 70 |
|
| 71 |
-
While the model demonstrates an accuracy of
|
| 72 |
|
| 73 |
### Recommendations
|
| 74 |
|
| 75 |
Users (both direct and downstream) should be made aware of the risks, biases, and limitations of the model. Specifically:
|
| 76 |
|
| 77 |
* **Domain Verification:** Since EinsteinNet is optimized for scientific and engineering imaging (as detailed in [EinsteinNet and state-of-the-art ML models for android-based orange classification: Integration, evaluation, and deployment](https://doi.org/10.1016/j.atech.2025.101072)), users should perform domain-specific validation before deploying it for critical decision-making.
|
| 78 |
-
* **Edge Deployment:** Given its small footprint (
|
| 79 |
* **Input Consistency:** To minimize bias and risk, ensure input images are preprocessed to match the training resolution and normalization parameters used during development.
|
| 80 |
* **Human-in-the-loop:** This model should complement, not replace, expert analysis in high-stakes environments like medical diagnostics or thermal structural safety.
|
| 81 |
|
|
@@ -112,7 +112,7 @@ The model was trained on the [Orange Fruit Image Dataset for Classification](htt
|
|
| 112 |
|
| 113 |
#### Preprocessing
|
| 114 |
|
| 115 |
-
- **Resolution:** All images were resized to
|
| 116 |
- **Splitting:** The dataset was partitioned using stratified sampling:
|
| 117 |
- **Training:** 70% (10,500 images)
|
| 118 |
- **Validation:** 15% (2,250 images)
|
|
@@ -133,7 +133,7 @@ The model was trained on the [Orange Fruit Image Dataset for Classification](htt
|
|
| 133 |
- **Total Training Time:** 13,033.01 Seconds.
|
| 134 |
- **Model Size:** 2.54 MB.
|
| 135 |
- **Quantized Size:** 254 KB (Optimized for mobile/IoT deployment via LiteRT).
|
| 136 |
-
- **Inference Performance:** Real-world on-device accuracy remains
|
| 137 |
|
| 138 |
## Evaluation
|
| 139 |
---
|
|
@@ -192,7 +192,6 @@ The model architecture was specifically optimized to reduce the number of traina
|
|
| 192 |
|
| 193 |
## Environmental Impact
|
| 194 |
---
|
| 195 |
-
Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
|
| 196 |
|
| 197 |
- **Hardware Type:** GPU (Benchmarked for Mobile deployment on Google Pixel 6)
|
| 198 |
- **Hours used:** ~3.62 hours (13,033 seconds)
|
|
|
|
| 68 |
---
|
| 69 |
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
| 70 |
|
| 71 |
+
While the model demonstrates an accuracy of *99.6\%*, performance is strictly dependent on the consistency of the input data compared to the training set. Users should evaluate the model for bias if applying it to datasets with significantly different lighting or sensor noise profiles.
|
| 72 |
|
| 73 |
### Recommendations
|
| 74 |
|
| 75 |
Users (both direct and downstream) should be made aware of the risks, biases, and limitations of the model. Specifically:
|
| 76 |
|
| 77 |
* **Domain Verification:** Since EinsteinNet is optimized for scientific and engineering imaging (as detailed in [EinsteinNet and state-of-the-art ML models for android-based orange classification: Integration, evaluation, and deployment](https://doi.org/10.1016/j.atech.2025.101072)), users should perform domain-specific validation before deploying it for critical decision-making.
|
| 78 |
+
* **Edge Deployment:** Given its small footprint (*2.54* MB), the model is highly recommended for mobile and IoT applications. However, ensure the target hardware supports the Keras/LiteRT runtime to maintain the reported *99.6\%* accuracy.
|
| 79 |
* **Input Consistency:** To minimize bias and risk, ensure input images are preprocessed to match the training resolution and normalization parameters used during development.
|
| 80 |
* **Human-in-the-loop:** This model should complement, not replace, expert analysis in high-stakes environments like medical diagnostics or thermal structural safety.
|
| 81 |
|
|
|
|
| 112 |
|
| 113 |
#### Preprocessing
|
| 114 |
|
| 115 |
+
- **Resolution:** All images were resized to *224 x 224* pixels.
|
| 116 |
- **Splitting:** The dataset was partitioned using stratified sampling:
|
| 117 |
- **Training:** 70% (10,500 images)
|
| 118 |
- **Validation:** 15% (2,250 images)
|
|
|
|
| 133 |
- **Total Training Time:** 13,033.01 Seconds.
|
| 134 |
- **Model Size:** 2.54 MB.
|
| 135 |
- **Quantized Size:** 254 KB (Optimized for mobile/IoT deployment via LiteRT).
|
| 136 |
+
- **Inference Performance:** Real-world on-device accuracy remains *>95\%* despite variability in lighting and backgrounds.
|
| 137 |
|
| 138 |
## Evaluation
|
| 139 |
---
|
|
|
|
| 192 |
|
| 193 |
## Environmental Impact
|
| 194 |
---
|
|
|
|
| 195 |
|
| 196 |
- **Hardware Type:** GPU (Benchmarked for Mobile deployment on Google Pixel 6)
|
| 197 |
- **Hours used:** ~3.62 hours (13,033 seconds)
|