TomTomTommi / HiNet

Official PyTorch implementation of "HiNet: Deep Image Hiding by Invertible Network" (ICCV 2021)
166 stars 35 forks source link

Hi, how to obtain your testing sets? #5

Closed ddghjikle closed 2 years ago

ddghjikle commented 2 years ago

Thanks a lot for sharing this excellent work. However, according to the paper, the testing datasets include DIV2K [1] testing dataset with 100 images at resolution 1024 × 1024, ImageNet [25] with 50,000 images at resolution 256 × 256, and COCO [17] dataset with 5,000 images at resolution 256 × 256. It seems that images in the testing datasets are randomly chosen from their original datasets. Therefore, could you please release the testing datasets used in your paper?

TomTomTommi commented 2 years ago

Thanks. As can be seen in https://github.com/TomTomTommi/HiNet/blob/4409a70a82edc23eef5f467bf3143d59a4ad8020/datasets.py#L71, images in the testing datasets are chosen in order from their original datasets. In addition, we adopt center crop in https://github.com/TomTomTommi/HiNet/blob/4409a70a82edc23eef5f467bf3143d59a4ad8020/datasets.py#L53. Therefore, the images are calculated by order, and you can obtain the testing datasets simply by running the codes.

ddghjikle commented 2 years ago

Thanks very much. By your comments, it means that what I need to do is to place standard DIV2K dataset, ImageNet dataset, and COCO dataset into corresponding path. Then the codes can automatically choose testing images from these datasets. Am I right?

TomTomTommi commented 2 years ago

Yes, your understanding is right.

ddghjikle commented 2 years ago

Thanks very much!