diff --git a/training/tools/dataset_to_encoding_script.py b/training/tools/dataset_to_encoding_script.py index d091bdbacae256edc390ac8e52856a1296ed210a..4258ca82236e59d580255984a343cdad9f88f47e 100644 --- a/training/tools/dataset_to_encoding_script.py +++ b/training/tools/dataset_to_encoding_script.py @@ -211,12 +211,17 @@ for seq in dataset: else echo "[CACHED] {basename}_QP{qp} already encoded" fi + if [ ! -f ${{DECDIR}}/{basename}_QP{qp}.log ]; then if [ -f ${{ENCDIR}}/{basename}_QP{qp}.bin -a -f ${{ENCDIR}}/{basename}_QP{qp}.log ]; then - $DEC -b ${{ENCDIR}}/{basename}_QP{qp}.bin --DumpBasename=${{DUMPDIR}}/{basename}_QP{qp} $OPTDEC > ${{DECDIR}}/{basename}_QP{qp}.log - echo "[INFO] {basename}_QP{qp} decoded" - else - echo "[INFO] skipped decoding (no bitstream)"; + if grep -q "Total Time" ${{ENCDIR}}/{basename}_QP{qp}.log; then + $DEC -b ${{ENCDIR}}/{basename}_QP{qp}.bin --DumpBasename=${{DUMPDIR}}/{basename}_QP{qp} $OPTDEC > ${{DECDIR}}/{basename}_QP{qp}.log + echo "[INFO] {basename}_QP{qp} decoded" + else + echo "[INFO] {basename}_QP{qp} encoding not finished"; + fi + else + echo "[INFO] {basename}_QP{qp} skipped decoding (no bitstream)"; fi; else echo "[CACHED] {basename}_QP{qp} already decoded";