Add VMAF support.

metrics_gather.sh

@@ -65,6 +65,14 @@ if [ -z "$DUMP_CIEDE" ]; then
   export DUMP_CIEDE="$DAALATOOL_ROOT/tools/dump_ciede2000.py"
 fi
 
+if [ -z "$VMAF_ROOT" ]; then
+  export VMAF_ROOT="$DAALATOOL_ROOT/../vmaf"
+fi
+
+if [ -z "$VMAFOSSEXEC" ]; then
+  export VMAFOSSEXEC="$VMAF_ROOT/wrapper/vmafossexec"
+fi
+
 if [ -z "$YUV2YUV4MPEG" ]; then
   export YUV2YUV4MPEG="$DAALATOOL_ROOT/tools/yuv2yuv4mpeg"
 fi
@@ -238,6 +246,27 @@ fi
 
 echo "$ENCTIME"
 
+rm -f ref dis
+mkfifo ref
+mkfifo dis
+FORMAT=yuv420p
+case $CHROMA in
+420p10)
+  FORMAT=yuv444p10le
+  ;;
+444p10)
+  FORMAT=yuv444p10le
+  ;;
+444)
+  FORMAT=yuv444p
+  ;;
+esac
+"$DAALATOOL_ROOT/tools/y4m2yuv" "$FILE" -o ref &
+"$DAALATOOL_ROOT/tools/y4m2yuv" "$BASENAME.y4m" -o dis &
+VMAF=$("$VMAFOSSEXEC" $FORMAT $WIDTH $HEIGHT ref dis "$VMAF_ROOT/resource/model/nflxall_vmafv4.pkl" | tail -n 1)
+
+echo "$VMAF"
+
 if [ ! "$NO_DELETE" ]; then
   rm -f "$BASENAME.y4m" "$BASENAME.yuv" "$BASENAME.ogv" "$BASENAME.x264" "$BASENAME.x265" "$BASENAME.vpx" "$BASENAME.ivf" "$TIMEROUT" "$BASENAME-enc.out" "$BASENAME-psnr.out" "$BASENAME.thor" 2> /dev/null
 fi

work.py

@@ -73,6 +73,7 @@ def reduce(self):
                 f.write(str(work.metric['apsnr'][2])+' ')
                 f.write(str(work.metric['msssim'][0])+' ')
                 f.write(str(work.metric['encodetime'])+' ')
+                f.write(str(work.metric['vmaf'])+' ')
                 f.write('\n')
                 f.close()
             else:
@@ -138,6 +139,7 @@ def parse(self, stdout, stderr):
             self.metric['msssim'][1] = split[50]
             self.metric['msssim'][2] = split[52]
             self.metric['encodetime'] = split[53]
+            self.metric['vmaf'] = split[57]
             self.failed = False
         except IndexError:
             rd_print(self.log,'Decoding result for '+self.filename+' at quality '+str(self.quality)+' failed!')