We provide installation instructions for ImageNet classification experiments here.
Create an new conda virtual environment
conda create -n convnext python=3.8 -y
conda activate convnext
Install Pytorch>=1.8.0, torchvision>=0.9.0 following official instructions. For example:
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html
Clone this repo and install required packages:
git clone https://github.com/facebookresearch/ConvNeXt
pip install timm==0.3.2 tensorboardX six
The results in the paper are produced with torch==1.8.0+cu111 torchvision==0.9.0+cu111 timm==0.3.2
.
Download the ImageNet-1K classification dataset and structure the data as follows:
/path/to/imagenet-1k/
train/
class1/
img1.jpeg
class2/
img2.jpeg
val/
class1/
img3.jpeg
class2/
img4.jpeg
For pre-training on ImageNet-22K, download the dataset and structure the data as follows:
/path/to/imagenet-22k/
class1/
img1.jpeg
class2/
img2.jpeg
class3/
img3.jpeg
class4/
img4.jpeg