mmdetection使用第二章显卡训练
时间: 2023-12-04 19:09:17 浏览: 90
MMDetection实战:MMDetection训练与测试
5星 · 资源好评率100%
MMDetection是一个基于PyTorch实现的目标检测开源项目,由商汤和港中文大学共同推出。它提供了许多目标检测算法的实现,并封装了数据集构建、模型搭建和训练策略等过程为模块,使得使用者能够以较少的代码量来训练和测试目标检测模型。
在MMDetection中进行训练时,你可以选择使用不同的硬件设备,比如Nvidia GeForce RTX 2080Ti显卡。而在训练阶段,可以将ImageNet 2012数据集的训练图像划分为子训练集和子验证集。子验证集包含从训练图像中随机选择的图像,用于评估剪枝网络在子验证数据集上的性能。剩余的图像构成子训练集,用于训练权重生成元网络。
因此,MMDetection可以使用第二章提到的Nvidia GeForce RTX 2080Ti显卡进行训练,并通过ImageNet 2012数据集的子训练集和子验证集进行训练和评估。<span class="em">1</span><span class="em">2</span><span class="em">3</span><span class="em">4</span>
阅读全文