rd_collect_x265.sh 1.4 KB

1234567891011121314151617181920212223242526272829303132333435
  1. #!/bin/bash
  2. set -e
  3. if [ -z $RD_COLLECT_SUB ]; then
  4. echo "Please use: $(dirname $0)/rd_collect.sh x265 *.y4m"
  5. exit 1
  6. fi
  7. FILE=$1
  8. BASENAME=$(basename $FILE)-$CODEC
  9. rm $BASENAME.out 2> /dev/null || true
  10. echo $BASENAME
  11. WIDTH=$(head -1 $FILE | cut -d\ -f 2 | tr -d 'W')
  12. HEIGHT=$(head -1 $FILE | cut -d\ -f 3 | tr -d 'H')
  13. RANGE=$(seq 1 51)
  14. QSTR="--preset slow --frame-threads 1 --min-keyint 256 --keyint 256 --no-scenecut --crf=\$x"
  15. for x in $RANGE; do
  16. $X265 -r $BASENAME.y4m $(echo $QSTR | sed 's/\$x/'$x'/g') -o $BASENAME.x265 $FILE 2> $BASENAME-$x-enc.out > /dev/null
  17. SIZE=$(stat -c %s $BASENAME.x265)
  18. $DUMP_PSNR $FILE $BASENAME.y4m > $BASENAME-$x-psnr.out 2> /dev/null
  19. FRAMES=$(cat $BASENAME-$x-psnr.out | grep ^0 | wc -l)
  20. PIXELS=$(($WIDTH*$HEIGHT*$FRAMES))
  21. PSNR=$(cat $BASENAME-$x-psnr.out | grep Total | tr -s ' ' | cut -d\ -f $((4+$PLANE*2)))
  22. PSNRHVS=$($DUMP_PSNRHVS $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\ -f $((4+$PLANE*2)))
  23. SSIM=$($DUMP_SSIM $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\ -f $((4+$PLANE*2)))
  24. FASTSSIM=$($DUMP_FASTSSIM -c $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\ -f $((4+$PLANE*2)))
  25. rm $BASENAME.x265 $BASENAME.y4m $BASENAME-$x-enc.out $BASENAME-$x-psnr.out
  26. echo $x $PIXELS $SIZE $PSNR $PSNRHVS $SSIM $FASTSSIM >> $BASENAME.out
  27. #tail -1 $BASENAME.out
  28. done