cong-yang / MoireDet

85 stars 8 forks source link

Excellent Work #2

Open daeing opened 1 year ago

daeing commented 1 year ago

Thanks for your excellent work. Can you upload the code how to use moire layer and scene image to get moire scene image first? The synthetic dataset can upload latter. We can generate synthetic dataset by ourself.

daeing commented 1 year ago

作者您好,我看到您今天更新了如何生成摩尔纹的图片。然后我跑了一下结果,对于您生成合成摩尔纹代码我有一些想要和您确认的地方。 moire_patterns_dir = '/data/cong/moire/final/layers_ori_pattern' 您的这个 moire_patterns_dir 参数没有用上。但是您的注释描述是#Moire pattern means the extracted moire's edge By Cong Yang,看起来像是先获取了摩尔纹的edge。

def split_moire_datasets(src_dir, train_dir, test_dir, train_ratio=0.8): os.makedirs(train_dir, exist_ok=True)

os.makedirs(train_dir.replace('layers_ori', 'layers_ori_pattern'), exist_ok=True)

os.makedirs(test_dir, exist_ok=True)

os.makedirs(test_dir.replace('layers_ori', 'layers_ori_pattern'), exist_ok=True)

imgs = [os.path.join(src_dir, v) for v in os.listdir(src_dir)]

random.shuffle(imgs)

train_imgs = imgs[:int(len(imgs) * train_ratio)]
eval_imgs = imgs[int(len(imgs) * train_ratio):]

for img in train_imgs:
    shutil.copy(
        img.replace('layers_ori', 'layers_ori_pattern'),
        train_dir.replace('layers_ori', 'layers_ori_pattern'))
    shutil.copy(img, train_dir)

for img in eval_imgs:
    shutil.copy(
        img.replace('layers_ori', 'layers_ori_pattern'),
        test_dir.replace('layers_ori', 'layers_ori_pattern'))
    shutil.copy(img, test_dir)

然后上面这个函数执行的时候是将moire原图分别拷贝到了layer_ori和layers_ori_pattern中了,但是和您上面的注释是冲突的?

所以我想知道layer_ori存储的应该是拍摄的摩尔纹图片,layers_ori_pattern应该存储的是使用你仓库中的这个文件moirelayer.py 得到的边界是嘛?

非常感谢您开源您的工作,祝您生活开心。

Thothene commented 1 year ago

您好,请问您有这篇文章的MoireIDT数据集吗,我在下载的时候被提示链接失效了,如果有的话是否方便给我一份,谢谢!

A-biao96 commented 1 year ago

您好,请问您有这篇文章的MoireIDT数据集吗,我在下载的时候被提示链接失效了,如果有的话是否方便给我一份,谢谢!

把链接后面的pwd=1234,去掉,手动输入密码,例如:https://pan.baidu.com/s/1JExFI9-uesMfhgMnr5GWLQ

zhangwen118 commented 4 days ago

@daeing 是的layers_ori_pattern 是作者moirelayer.py这个代码用拉普拉斯算子在白屏的摩尔纹图中提取的,你应该好好读读作者的论文,非常棒的工作。。。