
pyTorch
Awiny_ImageProcess
没有当Engineer的实力,也要有AdEngineer的追求。
展开
-
监控场景下的行人视觉计算平台
监控场景下的行人视觉计算平台_v0简易版的行人计算机视觉平台, 支持监控摄像头下行人/车辆/人脸检测;行为识别/车辆属性分析/行人属性分析;原创 2020-09-19 14:49:58 · 295 阅读 · 0 评论 -
pyTorch 使用多GPU训练
1.在pyTorch中模型使用GPU训练很方便,直接使用model.gpu()。 2.使用多GPU训练,model = nn.DataParallel(model) 3.注意训练/测试过程中 inputs和labels均需加载到GPU中 inputs, labels = Variable(inputs.cuda()), Variable(labels.cuda())具体使用参考 pyto...原创 2018-04-16 17:33:24 · 8675 阅读 · 0 评论