Mark
/
celeste-ai
Archived
1
0
Fork 0
This repository has been archived on 2023-11-28. You can view files and clone it, but cannot push or open issues/pull-requests.
celeste-ai/celeste/ffmpeg.sh

91 lines
1.6 KiB
Bash

#!/bin/bash
# Where screenshots are saved.
# SC_ROOT/SC_DIR should contain episode screenshot directories
SC_ROOT="model_data/current"
SC_DIR="screenshots"
# Select a temporary working directory
# if false, uses ramdisk.
# set to true if ramdisk overflows.
if false; then
OUTPUT_DIR="model_data/video_output"
# This directory will be deleted.
# Make sure it doesn't already exist.
if [ -e "$OUTPUT_DIR" ]; then
echo "$OUTPUT_DIR exists, exiting. Please delete it manually."
exit 1
fi
mkdir -p $OUTPUT_DIR
else
OUTPUT_DIR=$(mktemp -d)
fi
# Usage:
# render_episode <src_dir> <output_name>
#
# Turns a directory of frame screenshots into a video.
# Applies upscaling. We do it early, because upscaling
# after encoding will exaggerate artifacts.
render_episode () {
ffmpeg \
-y \
-loglevel quiet \
-framerate 30 \
-i "$1/hackcel_%003d.png" \
-c:v libx264 \
-crf 20 \
-preset slow \
-tune animation \
-vf "scale=512x512:flags=neighbor" \
"$2.mp4"
}
echo "Making episode files..."
for D in "$SC_ROOT/$SC_DIR"/*; do
if [ -d "${D}" ]; then
render_episode "$D" "$OUTPUT_DIR/${D##*/}"
fi
done
echo "Merging..."
for f in "$OUTPUT_DIR"/*.mp4; do
echo file \'$f\' >> "$OUTPUT_DIR/video_merge_list"
done
# Merge videos
ffmpeg \
-loglevel error -stats -y \
-f concat \
-safe 0 \
-i "$OUTPUT_DIR/video_merge_list" \
"$SC_ROOT/1x.mp4"
echo ""
echo "Making accelerated video..."
# Make accelerated video
ffmpeg \
-loglevel error -stats -y \
-i "$SC_ROOT/1x.mp4" \
-framerate 60 \
-filter:v "setpts=0.125*PTS" \
"$SC_ROOT/8x.mp4"
echo "Cleaning up...."
rm -dr $OUTPUT_DIR