Skip to content

Latest commit

 

History

History
18 lines (11 loc) · 1.01 KB

流水账.md

File metadata and controls

18 lines (11 loc) · 1.01 KB

操作流程手册

1 数据集的搜集、处理等

小米8和iphone所拍摄的原始照片分辨率很高,达到4000*3000的级别,先裁剪去掉水印,再resized_img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)采样为原来分辨率的四分之一。得到input1和input2

2 Warp

使用udis的warp部分进行warp,input1和input2输入udis2的warp,进行warp,生成mask1、mask2、warp1、warp2;提供初步的形变mask

3 salient object mask

运行selformer生成salient object mask;由于生成的object mask是512*512;需要参照步骤2中生成的mask1(或者mask2)进行resize;这个步骤中的resize代码在/public/home/jinjiping2023/OASD/oasd/data-processed.ipynb中

4 Seam detection

使用我们预先训练好的改动了object loss的Seam detection模型在步骤二的warp的mask上进行进一步的接缝预测;并将结果保存。

魔改网络

可以使用一些,dilated convolution来降低时间复杂度,最后比较一下运行时间