search

liuqk3 / PUT

Paper 'Transformer based Pluralistic Image Completion with Reduced Information Loss' in TPAMI 2024 and 'Reduce Information Loss in Transformers for Pluralistic Image Inpainting' in CVPR2022
MIT License
173 stars 15 forks source link

推理问题 #25

Open zhangshilong2000 opened 10 months ago

zhangshilong2000 commented 10 months ago

您好，我在使用命令：python scripts/inference.py --name transformer_mural --func inference_complet_sample_in_feature_for_evaluation --gpu 0 --batch_size 2 修复单个图像时，出现如下问题： 输入带有掩码的图像：image 修复的图像：image

liuqk3 commented 8 months ago

这种情况感觉不太对。建议换成inference_inpainting.py这个脚本进行测试，相对比较简单且灵活，具体可以看看readme。

zmzzz12345 commented 4 months ago

您好，我在使用命令：python scripts/inference.py --name transformer_mural --func inference_complet_sample_in_feature_for_evaluation --gpu 0 --batch_size 2 修复单个图像时，出现如下问题： 输入带有掩码的图像：image 修复的图像：image 你好，请问执行单张图像推理需要修改哪些参数呢？我用这个命令训练，结果是找不到文件，我的图片大小是160*160 的python scripts/inference_inpainting.py --func inference_inpainting --name D:/zm/transformer_ffhq/checkpoint/last.pth --input_res 160,160 --n um_token_per_iter 1 --num_token_for_sampling 50 --image_dir ./HR --mask_dir ./mask --save_masked_image --save_dir E:/zm/predict/mask --num_sample 1 --gpu 0