brain-diffuser / vdvae /setup_imagenet.sh
dineshsai07's picture
Add files using upload-large-folder tool
46a8d8a verified
if [ "$1" == "imagenet32" ]; then
echo "downloading imagenet32"
wget http://www.image-net.org/small/train_32x32.tar
wget http://www.image-net.org/small/valid_32x32.tar
tar -xvf train_32x32.tar
tar -xvf valid_32x32.tar
python files_to_npy.py train_32x32/ imagenet32-train.npy
python files_to_npy.py valid_32x32/ imagenet32-valid.npy
elif [ "$1" == "imagenet64" ]; then
echo "downloading imagenet64"
wget http://www.image-net.org/small/train_64x64.tar
wget http://www.image-net.org/small/valid_64x64.tar
tar -xvf train_64x64.tar
tar -xvf valid_64x64.tar
python files_to_npy.py train_64x64/ imagenet64-train.npy
python files_to_npy.py valid_64x64/ imagenet64-valid.npy
else
echo "please pass the string imagenet32 or imagenet64 as an argument"
fi