MikkoLipsanen
commited on
Commit
•
32043f4
1
Parent(s):
375fd17
Update README.md
Browse files
README.md
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
-
---
|
2 |
-
pipeline_tag: image-classification
|
3 |
-
---
|
4 |
## Table cell classification
|
5 |
|
6 |
The model is trained to classify table cell images as either empty or not empty. It has been trained using
|
@@ -29,7 +29,9 @@ Validation dataset consisted of
|
|
29 |
## Training procedure
|
30 |
|
31 |
The code used for model training is available in the repository in `train.py` file, which uses functions from
|
32 |
-
`augment.py` and `utils.py` files. The
|
|
|
|
|
33 |
|
34 |
- image size: 2560
|
35 |
- learning rate: 0.0001
|
|
|
1 |
+
---
|
2 |
+
pipeline_tag: image-classification
|
3 |
+
---
|
4 |
## Table cell classification
|
5 |
|
6 |
The model is trained to classify table cell images as either empty or not empty. It has been trained using
|
|
|
29 |
## Training procedure
|
30 |
|
31 |
The code used for model training is available in the repository in `train.py` file, which uses functions from
|
32 |
+
`augment.py` and `utils.py` files. The required libraries are listed in the `requirements.txt` file.
|
33 |
+
|
34 |
+
The model was trained using cpu with the following hyperparameters:
|
35 |
|
36 |
- image size: 2560
|
37 |
- learning rate: 0.0001
|