pytorch
Mr_Yoky
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
import open3d时报Error in `python‘: free(): invalid pointer: 0x000055
原引用是这样的, torch在open3d之前引用,结果报错import numpy as npimport osimport pickleimport torchimport torch.utils.data as torch_dataimport lib.utils.kitti_utils as kitti_utilsimport lib.utils.roipool3d.roipool3d_utils as roipool3d_utilsfrom lib.config import原创 2021-03-31 11:48:38 · 919 阅读 · 0 评论 -
关于pytorch的Dataloader和Datasets类的理解
关于Dataloader和Datasets类的理解,示例代码:继承自torch_data.Dataset类,其中的 __len__(self)和__getitem__(self,index)方法必须重写,在使用DataLoader加载数据时,会根据__len__(self)的长度作为遍历的上限,实际返回的个数为__len__(self)除以batch_size,同时返回的结果为调用__getitem__(self,index)这个函数batch_size次,然后拼接的结果,返回的数据可以是各种类型,而具原创 2021-03-17 14:03:13 · 475 阅读 · 0 评论
分享