img2float / README.md
Arigadam's picture
Update README.md
f7a582c verified
---
library_name: keras
license: mit
---
## Model description
A very simple model that converts an image into a number!
### the hepler function
(requirements: `numpy Pillow`)
```python
import numpy as np
from PIL import Image
def predict(model, img):
pil_image = img
pil_image = pil_image.resize((64, 64))
image_array = np.array(pil_image) / 255.0
image_array = np.expand_dims(image_array, axis=0)
input_shape = (64, 64, pil_image.mode == 'RGB' and 3 or 1)
decimal_prediction = model.predict(image_array)[0][0]
return decimal_prediction
```
## Intended uses & limitations
More information needed
## Training and evaluation data
More information needed
## Training procedure
### Training hyperparameters
The following hyperparameters were used during training:
| Hyperparameters | Value |
| :-- | :-- |
| name | Adam |
| weight_decay | None |
| clipnorm | None |
| global_clipnorm | None |
| clipvalue | None |
| use_ema | False |
| ema_momentum | 0.99 |
| ema_overwrite_frequency | None |
| jit_compile | False |
| is_legacy_optimizer | False |
| learning_rate | 0.0010000000474974513 |
| beta_1 | 0.9 |
| beta_2 | 0.999 |
| epsilon | 1e-07 |
| amsgrad | False |
| training_precision | float32 |
## Model Plot
<details>
<summary>View Model Plot</summary>
![Model Image](./model.png)
</details>