Harden operator YOLO training runtime
This commit is contained in:
+2
-1
@@ -265,7 +265,7 @@ Export the same operator samples to a local YOLO detection dataset when the
|
||||
public model candidates are not strong enough for the target imagery:
|
||||
|
||||
```bash
|
||||
docker exec -it geointel python /app/scripts/export_operator_yolo_dataset.py \
|
||||
docker exec -it geointel python3 /app/scripts/export_operator_yolo_dataset.py \
|
||||
--manifest-path /app/storage/operator-data/operator_samples_manifest.json \
|
||||
--output-dir /app/storage/operator-data/yolo-building-dataset \
|
||||
--val-samples turnhout \
|
||||
@@ -291,6 +291,7 @@ docker exec \
|
||||
-e TRAIN_BATCH=2 \
|
||||
-e TRAIN_WORKERS=0 \
|
||||
-e TRAIN_DEVICE=cpu \
|
||||
-e PYTHON_BIN=python3 \
|
||||
geointel bash /app/scripts/train_operator_yolo_detector.sh
|
||||
```
|
||||
|
||||
|
||||
Reference in New Issue
Block a user