qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-10-07 06:59:58 +02:00
Code Issues Projects Releases Wiki Activity

Update README.md: how to train model using docker image

Browse source
This commit is contained in:
vahidrezanezhad 2025-06-25 18:33:55 +02:00 • committed by GitHub
parent f5a1d1a255
commit 1b222594d6
1 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
train/README.md
View file

@ -25,6 +25,18 @@ each class will be defined with a RGB value and beside images, a text file of cl
### Train ### Train
To train a model, run: ``python train.py with config_params.json`` To train a model, run: ``python train.py with config_params.json``
### Train using Docker
#### Build the Docker image
```bash
docker build -t model-training .
```
#### Run Docker image
```bash
docker run --gpus all -v /host/path/to/entry_point_dir:/entry_point_dir model-training
```
### Ground truth format ### Ground truth format
Lables for each pixel are identified by a number. So if you have a Lables for each pixel are identified by a number. So if you have a
binary case, ``n_classes`` should be set to ``2`` and labels should binary case, ``n_classes`` should be set to ``2`` and labels should