Which specific ImageNet dataset should be used to train, to reproduce the results of your experiments in the paper?

ToTheBeginning / GCP-Colorization

Official code for ICCV 2021 paper "Towards Vivid and Diverse Image Colorization with Generative Color Prior".

Apache License 2.0

44 stars 4 forks source link

Which specific ImageNet dataset should be used to train, to reproduce the results of your experiments in the paper? #1

Closed HyZhu39 closed 2 years ago

HyZhu39 commented 2 years ago

Thank you for your excellent work. I was greatly inspired after reading your paper and wanted to reproduce your experimental results, but I encountered a problem during training. The ImageNet dataset from the official website:

https://image-net.org/download-images.php

There are a lot of versions to choose from. Which ImageNet subset did you choose for your network? Since I don't know enough about the ImageNet dataset, this may be a stupid question, thanks, and looking forward to your answer.

ToTheBeginning commented 2 years ago

We use ImageNet-1k dataset which contains 1,281,167 training images and 50,000 validation images.