57 lines
1.3 KiB
Bash
57 lines
1.3 KiB
Bash
#!/bin/bash
|
|
|
|
# Initialize variables
|
|
prompts=()
|
|
output_root=""
|
|
|
|
# Parse arguments
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
--prompts)
|
|
shift
|
|
while [[ $# -gt 0 && ! "$1" =~ ^-- ]]; do
|
|
prompts+=("$1")
|
|
shift
|
|
done
|
|
;;
|
|
--output_root)
|
|
output_root="$2"
|
|
shift 2
|
|
;;
|
|
*)
|
|
echo "Unknown argument: $1"
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# Validate required arguments
|
|
if [[ ${#prompts[@]} -eq 0 || -z "$output_root" ]]; then
|
|
echo "Missing required arguments."
|
|
echo "Usage: bash run_text2asset3d.sh --prompts \"Prompt1\" \"Prompt2\" --output_root <path>"
|
|
exit 1
|
|
fi
|
|
|
|
# Print arguments (for debugging)
|
|
echo "Prompts:"
|
|
for p in "${prompts[@]}"; do
|
|
echo " - $p"
|
|
done
|
|
echo "Output root: ${output_root}"
|
|
|
|
# Concatenate prompts for Python command
|
|
prompt_args=""
|
|
for p in "${prompts[@]}"; do
|
|
prompt_args+="\"$p\" "
|
|
done
|
|
|
|
# Step 1: Text-to-Image
|
|
eval python3 embodied_gen/scripts/text2image.py \
|
|
--prompts ${prompt_args} \
|
|
--output_root "${output_root}/images"
|
|
|
|
# Step 2: Image-to-3D
|
|
python3 embodied_gen/scripts/imageto3d.py \
|
|
--image_root "${output_root}/images" \
|
|
--output_root "${output_root}/asset3d"
|