qurator-spk/sbb_pixelwise_segmentation
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git synced 2025-07-01 14:40:05 +02:00
Code Issues 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 ce4219e517
commit cf63bd92bc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

14
README.md
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