Dandellion/triav1lity
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

do vmaf calculation

Browse Source
This commit is contained in:
Daniel Olsen 2020-07-15 05:14:52 +02:00
parent ef043548c7
commit 4603768ac2
2 changed files with 18 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
default.nix
View File

@ -1,10 +1,10 @@
let let
pkgs = import <nixpkgs> {}; pkgs = import <nixos-unstable> {};
in in
with pkgs; with pkgs;
stdenv.mkDerivation { stdenv.mkDerivation {
pname = "triav1c"; pname = "triav1c";
version = "0.0.0-20200715-0"; version = "0.0.0-20200715-1";
src = ./. ; src = ./. ;
@ -16,6 +16,7 @@ stdenv.mkDerivation {
chmod +x $out/bin/triav1c chmod +x $out/bin/triav1c
wrapProgram $out/bin/triav1c \ wrapProgram $out/bin/triav1c \
--prefix PATH : ${lib.makeBinPath [ ffmpeg libaom ]} --prefix PATH : ${lib.makeBinPath [ ffmpeg-full libaom ]} \
--prefix MODEL_PATH : ${libvmaf}
''; '';
} }

27
src/static/triav1c.sh
View File

@ -25,14 +25,14 @@ encode_aomenc_two_pass() {
aom_options=${aom_options//[^a-zA-Z0-9_\- =]/} aom_options=${aom_options//[^a-zA-Z0-9_\- =]/}
# Same story as above but also # Same story as above but also
ffmpeg_options="$3" ffmpeg_options="$3"
ffmpeg_options=${ffmpego//[^a-zA-Z0-9_\- =:]/} ffmpeg_options=${ffmpeg_options//[^a-zA-Z0-9_\- =:]/}
# set to boolean # set to boolean
doVMAF=$4 doVMAF="$4"
set +e set +e
ffmpeg -nostats -hide_banner -loglevel warning \ eval '#ffmpeg -nostats -hide_banner -loglevel warning \
-i "$file" "$ffmpeg_options" -f yuv4mpegpipe - | aomenc - "$aom_options" \ -i '$file' $ffmpeg_options -f yuv4mpegpipe - | aomenc - '$aom_options' \
--pass=1 --passes=2 --fpf="$file".fpf --ivf -o "$file".out.ivf --pass=1 --passes=2 --fpf='$file'.fpf --ivf -o '$file'.out.ivf'
retval=$? retval=$?
if [[ $retval -ne 0 ]]; then if [[ $retval -ne 0 ]]; then
echo "Error running aomenc pass 1 of 2" >&2 echo "Error running aomenc pass 1 of 2" >&2
@ -41,9 +41,9 @@ encode_aomenc_two_pass() {
return 1 return 1
fi fi
ffmpeg -nostats -hide_banner -loglevel warning \ eval '#ffmpeg -nostats -hide_banner -loglevel warning \
-i "$file" "$ffmpeg_options" -f yuv4mpegpipe - | aomenc - "$aom_options" \ -i '$file' $ffmpeg_options -f yuv4mpegpipe - | aomenc - '$aom_options' \
--pass=1 --passes=2 --fpf="$file".fpf --ivf -o "$file".out.ivf --pass=2 --passes=2 --fpf='$file'.fpf --ivf -o '$file'.out.ivf'
retval=$? retval=$?
if [ $retval -ne 0 ]; then if [ $retval -ne 0 ]; then
echo "Error running aomenc pass 2 of 2" >&2 echo "Error running aomenc pass 2 of 2" >&2
@ -56,10 +56,10 @@ encode_aomenc_two_pass() {
# This probably needs to be improved as well, so that it scales # This probably needs to be improved as well, so that it scales
# and sets the framerate automatically. This will likely never # and sets the framerate automatically. This will likely never
# Actually get used though, so it's fine. # Actually get used though, so it's fine.
if [[ doVMAF -eq true ]]; then if [[ $doVMAF == "true" ]]; then
ffmpeg -nostats -hide_banner -loglevel warning \ ffmpeg -nostats -hide_banner \
-r 24 -i "$file".out.ivf -r 24 "$file" -filter_complex \ -r 24 -i $file.out.ivf -r 24 -i $file -filter_complex \
"[0:v][1:v]libvmaf=log_fmt=json:log_path=$file.vmaf.json" -f null - >/dev/null "[0:v][1:v]libvmaf=model_path=$MODEL_PATH/share/model/vmaf_v0.6.1.pkl:log_fmt=json:log_path=$file.vmaf.json" -f null - >/dev/null
retval=$? retval=$?
if [ $retval -ne 0 ]; then if [ $retval -ne 0 ]; then
echo "Error running VMAF scan" >&2 echo "Error running VMAF scan" >&2
@ -72,10 +72,11 @@ encode_aomenc_two_pass() {
set -e set -e
rm -f \ rm -f \
"$file" \ # "$file" \
"$file".fpf \ "$file".fpf \
"$file".vmaf.json "$file".vmaf.json
return 0 return 0
} }
encode_aomenc_two_pass "$1" "-b 10 --cpu-used=5 --end-usage=q --cq-level=50 --tile-columns=0 --tile-rows=0 --lag-in-frames=35 --auto-alt-ref=1" "" true