- 博客(4)
- 收藏
- 关注
原创 kaggle-狗的品种分类
数据集是train,test两个文件夹,所有种类的狗都放在一个文件夹里,我们要使用imagefolder函数就得将train文件夹分成120个子文件夹,因为共有120种类的狗。【1】重建数据集def mkdir_if_not_exist(path): if not os.path.exists(os.path.join(*path)): os.makedirs(os.path.join(*path))构建文件夹函数,因为每一个种类的狗都需新建一个文件夹。labels = pd
2020-07-07 16:41:27
679
原创 pytorch构建自己的数据集
ImageFolder可以使用的情况是,比如猫狗识别中,train一个文件夹下已经将猫和狗分为两个不同的文件夹了,那我们可以直接使用ImageFolder来包装成数据集。但有时候遇到的数据集是train一个文件夹下包含了所有的猫狗图片,那我们就无法使用ImageFolder函数,此时我们可以自己构造一个dataset类。核心是getitem函数,这个函数的主要功能是根据样本的索引,返回索引对应的一张图片的图像数据X与对应的标签Y,也就是返回一个对应的训练样本。具体而言,getitem的实现思路比较简单
2020-07-02 17:26:14
2719
原创 pytorch保存训练模型以及加载模型
【1】训练中保存网络参数model_save_dir = 为你要保存的位置torch.save(net.state_dict(), model_save_dir)【2】测试中加载保存的网络net.load_state_dict(torch.load(‘model_save_dir’))
2020-07-02 17:07:44
613
原创 linux终端快捷指令打开pycharm
linux终端快捷指令打开pycharm步骤1: vim ~/.bashrc2: pycharm = alias pycharm=“bash /home/lijuncheng/pycharm/pycharm-community-2020.1.1/bin/pycharm.sh” #输入自己pycharm.sh的目录3.source ~/.bashrc执行完之后就可以在linux终端输入pycharm命令来打开pycharm了...
2020-06-28 15:52:41
2783
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人