主流的深度学习框架: caffe, tensorflow, pytorch/caffe2, keras。
mxnet, paddlepaddle, theano?,cntk。
此外,基本所有的框架都支持CUDA
框架的使用场景:关于框架的选择,首先必须要学会tensorflow/pytorch,这两个框架是开发者用的最多,最丰富的
移动端算法 使用Caffe
掌握开源框架的步骤:
1 掌握数据的准备和使用
2 掌握模型的定义
3 掌握训练过程、数据可视化
4 掌握训练方法 测试方法
caffe:
tensorflow:
参考文档【tensorflow速成】Tensorflow图像分类从模型自定义到测试
1 创建计算图
2 运行会话