
模型
timertimer
这个作者很懒,什么都没留下…
展开
-
torch.gather scatter
torch.gatherimport torchb = a.gather(dim, index)b = torch.gather(src, 1, torch.tensor([[0,0],[1,0]]))dim:指定轴方向,定义了填充方式。对于二维张量,dim=0表示逐列进行行填充,而dim=1表示逐列进行行填充。* 当dim=1时,index[0][0]的元素是1,那么它想要查找a[0][1]中的元素;* 当dim=0时,index[0][0]的元素是1,那么它想查找的a[1][0]中的元素;原创 2022-03-28 19:48:00 · 202 阅读 · 0 评论 -
pytorch使用tensorboard
……1、安装pip install tensorboardX2、调用1) lossfrom tensorboardX import SummaryWriter# before trainlog_writer = SummaryWriter('log_file_path')# in traininglog_writer.add_scalar('Train/Loss', loss.data[0], niter) # in pytorch1.0 loss.data[0] should be原创 2022-02-18 17:02:07 · 445 阅读 · 0 评论 -
训练过程_
1.数据准备1.1KITTI下载并准备数据集mkdir ./data/kitti/ && mkdir ./data/kitti/ImageSets# Download data splitwget -c https://raw.githubusercontent.com/traveller59/second.pytorch/master/second/data/ImageSets/test.txt --no-check-certificate --content-disposit原创 2021-07-11 21:45:15 · 200 阅读 · 0 评论 -
数据集___
KITTI1 数据采集平台KITTI数据采集平台包括2个灰度摄像机,2个彩色摄像机,一个Velodyne 3D激光雷达,4个光学镜头,以及1个GPS导航系统。传感器:1惯性导航系统(GPS / IMU):OXTS RT 30031台激光雷达:Velodyne HDL-64E2台灰度相机,1.4百万像素:Point Grey Flea 2(FL2-14S3M-C)2个彩色摄像头,1.4百万像素:Point Grey Flea 2(FL2-14S3C-C)4个变焦镜头,4-8毫米:Edmund原创 2021-07-11 20:14:18 · 124 阅读 · 0 评论 -
mmdet3d
mmdetection3d安装a. Create a conda virtual environment and activate it.conda create -n open-mmlab python=3.7 -ysource activate open-mmlabb. Install PyTorch and torchvision following the official instructions.conda install pytorch==1.5.0 cudatoolkit=10.原创 2021-07-08 22:21:41 · 750 阅读 · 0 评论