yingchen001/BAT-Fill

Preprocess of mask

Opened this issue · 13 comments

Hi, @yingchen001

Thank you for sharing your nice work!

I have two questions about a preprocess of irregular mask dataset.

  1. Did you use NVIDIA Irregular Mask Dataset: Training Set and NVIDIA Irregular Mask Dataset: Testing Set in official page, right?

  2. In the above page, authors said that "please use random augmentation tricks". However, I cannot find augmentation for original masks in https://github.com/yingchen001/BAT-Fill/blob/master/data/inpaint_dataset.py#L187. In fact, I obtain a image as shown in below.
    1

Would you tell me these answers?

Best regards.

2
Actually, when I use a test set, I have no problem.

This question of training mask is not intrinsically related to your work, but could you please tell me?

Thanks

Hi,

Thanks for your interest.

  1. We're using the test set of the irregular masks.
  2. We apply random flip on both image and masks during training, as per our codes:
    if self.opt.phase == 'train' and not if_tran_sample: img = transforms.RandomHorizontalFlip()(img) img = transforms.ColorJitter(0.05, 0.05, 0.05, 0.05)(img) mask = transforms.RandomHorizontalFlip()(mask) mask = mask.filter(ImageFilter.MaxFilter(3))

Regards.

Thank you for your reply.

You mean you use the test set for training your model, right?

Yes, that's correct.

@yingchen001
您好,非常感谢您的分享,麻烦问一下,您的这个模型是将places2的数据直接作为输入的吗?我训练的时候只有这样一个显示
2022-03-03 20-48-49 的屏幕截图
训练完之后只有这样两个文件。
2022-03-03 20-59-55 的屏幕截图
是哪里出问题了吗?

Nope, that's all right. The file records the training iterations for continued training, it's okay to train without it.

非常感谢您的回复!
我这里又出现了新的问题:
1、这个只能训练一个epoch吗,我训练了1个epoch就停了
2022-03-04 09-05-26 的屏幕截图
2、有关于resume的操作吗?我直接运行的话没有读入原来的数据
B15D8E0F77E05AF8797AD7E25801B34A

而且这个tran_iter.txt文件里面是空的

2022-03-04 09-16-37 的屏幕截图

还有个不懂的地方,这里说的预训练权重是在哪里用到的,在训练过程中好像并没有用到这个
2022-03-04 09-36-09 的屏幕截图

  1. It will load the saved checkpoint if you want to continue training.
  2. As per error says, you may need to convert the tensor to CPU by using .cpu()
  3. The pretrained VGG model is used for training the upsampler, not the transformer.

Thanks.

cluster = torch.from_numpy(np.load('/data/vdc/yingchen.yu/project/GPT_Inpaint/kmeans_centers.npy'))
您好,我遇到一个问题,请问这个kmeans_centers.npy文件是需要自己下载,然后放到对应的路径吗

非常感谢您的回复! 我这里又出现了新的问题: 1、这个只能训练一个epoch吗,我训练了1个epoch就停了 2022-03-04 09-05-26 的屏幕截图 2、有关于resume的操作吗?我直接运行的话没有读入原来的数据 B15D8E0F77E05AF8797AD7E25801B34A

而且这个tran_iter.txt文件里面是空的

2022-03-04 09-16-37 的屏幕截图

你好,我也遇到了这个问题,请问你解决了吗。这些类似tran_iter.txt的文件需要自己先建立才会写入吗?还是会自动创建这个文件?

kmeans_centers

yes, please change the abs path to the downloaded kmeans path

非常感谢您的回复! 我这里又出现了新的问题: 1、这个只能训练一个epoch吗,我训练了1个epoch就停了 2022-03-04 09-05-26 的屏幕截图 2、有关于resume的操作吗?我直接运行的话没有读入原来的数据 B15D8E0F77E05AF8797AD7E25801B34A
而且这个tran_iter.txt文件里面是空的
2022-03-04 09-16-37 的屏幕截图

你好,我也遇到了这个问题,请问你解决了吗。这些类似tran_iter.txt的文件需要自己先建立才会写入吗?还是会自动创建这个文件?

they'll be auto created, it's empty because the training is stopped