File size: 914 Bytes
497c818 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | #!/usr/bin/env bash
set -euo pipefail
# Visual quality (FID / CLIP-text / CLIP-img / LPIPS) for one generated folder.
# GEN=outputs/infer_seg NAME=ours_seg SUFFIX=seg bash scripts/eval_visual_quality.sh
# SUFFIX is the single-control suffix (seg/edge/depth); leave empty for <stem>.png.
source "$(dirname "${BASH_SOURCE[0]}")/_env.sh"
GEN="${GEN:?set GEN=<generated image folder>}"
NAME="${NAME:-$(basename "${GEN}")}"
SUFFIX="${SUFFIX:-}"
OUT="${OUT:-${PKG_ROOT}/outputs/visual_quality_${NAME}.json}"
DEVICE="${DEVICE:-cuda:0}"; BATCH="${BATCH:-16}"
METRICS="${METRICS:-fid clip_text clip_img lpips}"
cd "${PKG_ROOT}"
python eval/eval_visual_quality.py \
--gen_dir "${GEN}" --name "${NAME}" \
${SUFFIX:+--control_suffix "${SUFFIX}"} \
--image_root "${EVAL_IMAGE_ROOT}" \
--clip_model "${CLIP_MODEL}" \
--metrics ${METRICS} \
--batch_size "${BATCH}" --device "${DEVICE}" \
--output_json "${OUT}"
|