这是2018年cvpr的一篇文章.作者当中提供了源码,地址如下:
https://github.com/zhirongw/lemniscate.pytorch
首先要安装pytorch,建议使用清华的镜像,否则特别慢,教程地址如下:
https://blog.csdn.net/qq_43237588/article/details/89183303
接着就可以运行代码了,在ImageNet上训练模型需要输入很多的运行参数,在CIFAR10可以直接运行,也就是直接运行cifar.py文件.但是运行的同学们都会发现一个问题,他需要在线下载数据集,非常的慢,小编今日提供一个本地运行的方法(以cifar10为例):
首先下载cifar10数据集,地址如下:
https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
然后在项目目录上建立data文件夹,将数据集放进去
然后再注释掉库函数cifar.py(注意是库函数)的
download_url(self.url, self.root, self.filename, self.tgz_md5)