Skip to content

The inpainting effect on the first frame works fine, but from the first frame to the video, the inpainting effect appears very pool. #48

Description

@JayChen7777

Thnk you for your work! I have a small problem

109388395065_fps8.mp4

Here is my the first inpainting frame and groud_truth masked image.
Image
Image

Here is my cmd:
`
export CUDA_VISIBLE_DEVICES=0

model_path="/mmu_vcg2_wjc_ssd/xxx/models/zai-org/CogVideoX-5b-I2V"
num_inference_steps=50
guidance_scale=6.0
num_videos_per_prompt=1
dtype="bfloat16"

inpainting_branches=(
/mmu_vcg2_wjc_ssd/xxx/models/TencentARC/VideoPainter/VideoPainter/checkpoints/branch
)

id_adapter_resample_learnable_path=/mmu_vcg2_wjc_ssd/xxx/models/TencentARC/VideoPainter/VideoPainterID/checkpoints

lora_rank=256

inpainting_sample_ids=($(seq 0 0))

inpainting_frames=49
down_sample_fps=8
overlap_frames=0
image_or_video_path="/mmu_vcg2_wjc_ssd/xxx/models/TencentARC/VPBench/videovo/raw_video"

data_kind="test"
if [ "$data_kind" == "test" ]; then
inpainting_mask_meta="/mmu_vcg2_wjc_ssd/xxx/models/TencentARC/VPBench/pexels_videovo_test_dataset.csv"
elif [ "$data_kind" == "val" ]; then
inpainting_mask_meta="/mmu_vcg2_wjc_ssd/xxx/models/TencentARC/VPBench/pexels_videovo_val_dataset.csv"
elif [ "$data_kind" == "train" ]; then
inpainting_mask_meta="/mmu_vcg2_wjc_ssd/xxx/models/TencentARC/VPBench/pexels_videovo_train_dataset.csv"
else
echo "data_kind must be test or train or val"
exit 1
fi

prev_clip_weight=0.0

img_inpainting_model="/mmu_vcg2_wjc_ssd/xxx/models/black-forest-labs/FLUX.1-Fill-dev"

llm_model="gpt-4o"

dilate_size=0

output_base_path="./inp_fps${down_sample_fps}dilate${dilate_size}_${data_kind}_no_prompt_1003"

if [ "$llm_model" != "None" ]; then
output_base_path="${output_base_path}_${llm_model}"
fi

if [ "$overlap_frames" != "0" ]; then
output_base_path="${output_base_path}overlap${overlap_frames}"
fi

if [ ! -d "$output_base_path" ]; then
mkdir -p "$output_base_path"
chmod -R 777 "$output_base_path"
fi

for inpainting_branch in "${inpainting_branches[@]}"; do
for inpainting_sample_id in "${inpainting_sample_ids[@]}"; do
output_path="${output_base_path}/${inpainting_sample_id}.mp4"
python inpaint.py
--prompt "$prompt"
--model_path "$model_path"
--inpainting_branch "$inpainting_branch"
--output_path "$output_path"
--num_inference_steps "$num_inference_steps"
--guidance_scale "$guidance_scale"
--num_videos_per_prompt "$num_videos_per_prompt"
--dtype "$dtype"
--generate_type "i2v_inpainting"
--inpainting_mask_meta "$inpainting_mask_meta"
--inpainting_sample_id "$inpainting_sample_id"
--inpainting_frames "$inpainting_frames"
--image_or_video_path "$image_or_video_path"
--first_frame_gt
--replace_gt
--mask_add
--down_sample_fps $down_sample_fps
--overlap_frames $overlap_frames
--prev_clip_weight $prev_clip_weight
--img_inpainting_model $img_inpainting_model
--llm_model $llm_model
--dilate_size $dilate_size
done
done`

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions