qq_39502099的博客

看不懂的C++呜呜呜

前言害，我本来是Ubuntu18+win10的双系统的，结果昨天在装软件的时候直接把ubuntu搞崩了，直接很多软件被删掉了。加上我本身很讨厌ubuntu18的个别设定，索性直接重装了Ubuntu20。因为要运行DF-VO，所以要安装相应的环境，然而当我装好后，发现有一个bug各种不能解决，解决玩之后运行起来，但是效果完全不一样。各种方式都试过了，就是跟Ubuntu18的效果不一样。通过各种调参，也加深了理解，虽然部分效果很一般，但是整体还是可以的。现在是真的很糟心BUG2.1.2 上来就跟我说没

自己写自定义Dataset类的时候参考了 PyTorch中文文档Dataset规定所有的子类都必须包含有两个方法__len__和__getitem____len__ 提供了数据集的大小__getitem__ 支持整数索引，范围从0到len(self)-1以下是自己的一点小心得：__len__返回的可以是整个长度，也可以不是，只要你__getitem__能够解析item参数即可__getitem__需要传入一个item参数，作为索引，并返回相应的数据tansor，我看了有些Dataset基类

目录在这里前言简单介绍Show you the Code路径哦整份代码备注参考前言纯粹想学习一下torch的数据集类，可能后面会用到吧。简单介绍我们在训练过程中除了写模型、训练等程序外，还会用到数据加载，而官方就几个数据加载的类，不足以满足个人的需求我想要将个人的npy文件数据集加载进来，还有每一个npy文件对应了一个target，于是便想动手写个Dataset参考pytorch文档可以看到，所有其他数据集都会进行经子类化，且所有的子类都应该override__len__和__getitem__

项目场景：DF-VO是ICRA2020国际会议的开源项目，想着搞下来玩玩，这是项目的传送门两篇论文也附上：Visual Odometry Revisited: What Should Be Learnt?DF-VO: What Should Be Learnt for Visual Odometry? 运行：This code was tested with Python 3.6, CUDA 9.0, Ubuntu 16.04, and PyTorch-1.1.github里面提到的相关环

随便记录一下自己的学习过程train.pyimport torchfrom torch.utils.data import DataLoaderfrom torchvision import datasetsfrom torchvision import transformsfrom torch import nn, optimfrom lenet5 import Lenet5from resnet import ResNet18def main(): batch_size =