计算机视觉(斯坦福2017课程)

1. 主流深度学习框架

Caffe,Torch,Theano,Tensorflow,Keras,PyTorch等。

  • 优势:
    (1)框架能够让你轻松的构建和使用一个庞大的计算图
    (2)能够轻松的在计算图中计算梯度
    (3)能够高效的运行在GPU上
Caffe / Caffe2
Theano / TensorFlow

(1)写一些代码来定义计算图
(2)然后可以运行图模型,重复利用它
【深度学习】cs231n 计算机视觉(3)
训练这个网络:
【深度学习】cs231n 计算机视觉(3)
如上过程需要将权重不断的在CPU和GPU之间复制,因此一般不使用这种方式,而是将w也放在GPU处理。
代码如下:
【深度学习】cs231n 计算机视觉(3)

Torch / PyTorch

AlexNet,VGG, GoogleNet,ResNet等

相关文章: