WendongZh / SPL

[IJCAI'21] Code for Context-Aware Image Inpainting with Learned Semantic Priors,
53 stars 4 forks source link

How to get test_mask_index~ #12

Open 1997Jessie opened 1 year ago

1997Jessie commented 1 year ago

hi~I wanna ask about test_mask_index, Could you share with it?

WendongZh commented 1 year ago

This file is only used to determine the img-mask pairs during the testing stage. The main reason to use this file is to have a fair comparison with other approaches.

You can simply generate this file by yourself. such as:

import numpy as np 
index = np.random.choice(test_img_number, mask_img_number, replace=False)
1997Jessie commented 1 year ago

此文件仅用于在测试阶段确定 img 掩码对。使用此文件的主要原因是与其他方法进行公平的比较。

您可以自己生成此文件。如:

import numpy as np 
index = np.random.choice(test_img_number, mask_img_number, replace=False)

thank you very much! So is it ok that change replace=False into replace=True ?

1997Jessie commented 1 year ago

T_T... There is a new problem. When I use --with_test, it exits that one of the gpu is out of memory, I already modify --bs 1,but it doesn't work. So, you have other suggestions?

WendongZh commented 1 year ago

此文件仅用于在测试阶段确定 img 掩码对。使用此文件的主要原因是与其他方法进行公平的比较。 您可以自己生成此文件。如:

import numpy as np 
index = np.random.choice(test_img_number, mask_img_number, replace=False)

thank you very much! So is it ok that change replace=False into replace=True ?

Only if mask_img_number >= test_img_number, you can use replace=True

WendongZh commented 1 year ago

T_T... There is a new problem. When I use --with_test, it exits that one of the gpu is out of memory, I already modify --bs 1,but it doesn't work. So, you have other suggestions?

Turn it down. I mean directly remove '--with_test' and you can use the last checkpoint for test.