Skip to content
Snippets Groups Projects
Commit 3256804d authored by Franck Galpin's avatar Franck Galpin
Browse files

solve issue #61

parent 2746706f
No related branches found
No related tags found
No related merge requests found
......@@ -281,7 +281,7 @@ If you need to adapt the settings of your device for training, please edit the f
When ready, simply run:
```sh
python3 training_scripts/NN_Filtering_HOP/training/main.py --json_config training_scripts/NN_Filtering_HOP/config.json \
python3 training_scripts/NN_Filtering_HOP/training/main.py training_scripts/NN_Filtering_HOP/config.json \
training_scripts/NN_Filtering_HOP/model/model.json \
training_scripts/NN_Filtering_HOP/training/cfg/training_default.json \
training_scripts/NN_Filtering_HOP/training/cfg/stage2.json
......@@ -290,7 +290,7 @@ python3 training_scripts/NN_Filtering_HOP/training/main.py --json_config traini
#### 2. Convert model
The last ONNX model is converted into float SADL format.
```sh
python3 training_scripts/NN_Filtering_HOP/convert/to_sadl.py training_scripts/NN_Filtering_HOP/config.json \
python3 training_scripts/NN_Filtering_HOP/convert/to_sadl.py --json_config training_scripts/NN_Filtering_HOP/config.json \
--input_model stage2/training --output_model stage2/conversion
```
......@@ -380,7 +380,7 @@ This step is similar to II-A-4.
```sh
LIST="bvi bvi_valid tvd tvd_valid";
for DS in $LIST; do
python3 tools/concatenate_dataset.py --json_config training_scripts/NN_Filtering_HOP/config.json --input_dir_json stage3/encdec_${DS} --output_json stage3/encdec_${DS}
python3 tools/concatenate_dataset.py --json_config training_scripts/NN_Filtering_HOP/config.json --input_dir_json stage3/encdec_${DS} --input_dir_dataset_json stage2/yuv_${DS} --output_json stage3/encdec_${DS}
done
```
It will generate a unique dataset for each dataset in ["stage3"]["encdec"]["path"] from all individual datasets in ["stage3"]["encdec_xxx"]["path"]/["dump_dir"] and encoder logs in ["stage3"]["encdec_xxx"]["enc_dir"].
......@@ -416,7 +416,7 @@ python3 training_scripts/NN_Filtering_HOP/training/main.py --json_config traini
#### 2. Convert model
The last ONNX model is converted into float SADL format.
```sh
python3 training_scripts/NN_Filtering_HOP/convert/to_sadl.py training_scripts/NN_Filtering_HOP/config.json \
python3 training_scripts/NN_Filtering_HOP/convert/to_sadl.py --json_config training_scripts/NN_Filtering_HOP/config.json \
--input_model stage3/training --output_model stage3/conversion
```
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment