TensorFlow入门实战|第2周:彩色图片分类

我的环境:

  • 语言环境:Python3.6.5
  • 编译器:jupyter notebook
  • 深度学习环境:TensorFlow2.4.1

文章目录

  • 一、前期工作
    • 1. 设置GPU
    • 2. 导入数据
    • 3. 归一化
    • 4. 可视化
  • 二、构建CNN网络
  • 三、编译
  • 四、训练模型
  • 五、预测
  • 六、模型评估


一、前期工作

1. 设置GPU

如果使用的是CPU可以忽略这步

import tensorflow as tf
gpus = tf.config.list_physical_devices

你可能感兴趣的:(小白训练营,#,TensorFlow入门,tensorflow,分类,深度学习)