上传者: 42131705
|
上传时间: 2022-01-27 20:41:31
|
文件大小: 13.52MB
|
文件类型: -
解开标签分布以实现长尾视觉识别(CVPR 2021)
此代码库基于。
安装
conda create -n longtail pip python=3.7 -y
source activate longtail
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
pip install pyyaml tqdm matplotlib sklearn h5py tensorboard
训练
初赛
下载Places-LT的预训练caffe resnet152模型:请参考。
准备数据集:CIFAR-100,Places-LT,ImageNet-LT,iNaturalist 2018
请在下载这些数据集。
CIFAR-100培训
对于不平衡比为0.01的CIFAR-100,请使用LADE:
python main.py -