search

DanielCho-HK / ICME2023-Attention-Aware-Anime-Line-Drawing-Colorization

6 stars 2 forks source link

给出的文件似乎不全，是否可以提供完整版的代码与数据集？ #2

Open kebai633932 opened 7 months ago

kebai633932 commented 7 months ago

给出的文件似乎不全，是否可以提供完整版的代码与数据集？

DanielCho-HK commented 7 months ago

给出的文件似乎不全，是否可以提供完整版的代码与数据集？

您好，我們模型的代碼是train_our和model_our。數據集可以在kaggle上搜索anime pair data。我們使用的就是那個數據。

kebai633932 commented 7 months ago

您好，数据集是这个Anime Sketch Colorization Pair（https://www.kaggle.com/datasets/ktaebum/anime-sketch-colorization-pair/discussion）吗？对数据集需要进行什么处理？ 系统找不到指定的路径。: './self_recon/sketch'和‘./self_recon\gt’，是否可以提供文件？

DanielCho-HK commented 7 months ago

您好，数据集是这个Anime Sketch Colorization Pair（https://www.kaggle.com/datasets/ktaebum/anime-sketch-colorization-pair/discussion）吗？对数据集需要进行什么处理？ 系统找不到指定的路径。: './self_recon/sketch'和‘./self_recon\gt’，是否可以提供文件？

哈哈，明白你的意思了。這兩個路徑是我用來計算psnr或者其他指標的。你可以自定義任何你本機上的路徑，重點在於這兩個文件夾分別存放的是上色的結果，以及gt，我的目的是為了計算這兩個文件夾內所有文件相互的相似度或者其他指標。

DanielCho-HK commented 7 months ago

您好，数据集是这个Anime Sketch Colorization Pair（https://www.kaggle.com/datasets/ktaebum/anime-sketch-colorization-pair/discussion）吗？对数据集需要进行什么处理？ 系统找不到指定的路径。: './self_recon/sketch'和‘./self_recon\gt’，是否可以提供文件？

你可以看到項目中有兩個dataset腳本，分別是dataset.py和dataset_oldversion.py這兩個的區別在於前者是讀取兩個文件夾內的圖像，後者是只讀取一個文件夾內的圖像，然後進行一步crop操作。之所以有這兩個版本的原因在於，我們使用的這個數據集，你可以發現，他是將sketch和color image左右拼在一起的。我們需要有一本顯式的crop的操作，將sketch和color image分開存儲到兩個tensor變量中，用於後續的計算。

kebai633932 commented 7 months ago

谢谢