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

This commit is contained in:
vahidrezanezhad 2025-06-25 18:33:55 +02:00 committed by GitHub
parent ce4219e517
commit cf63bd92bc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -24,7 +24,19 @@ each class will be defined with a RGB value and beside images, a text file of cl
### Train
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
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