TensorFlow学习笔记-1--tf.data.Dataset 加载训练数据(kaggle 猫狗训练集):
由于技术不是很好,所以就写点博客来记录自己的 “学习” 历程。使用的是TensorFLow1.x。使用tf.data.Dataset来加载训练数据:
参考文章:https://www.jianshu.com/p/ceb9f9185b20
import cv2,os,csv
import tensorflow as tf
import matplotlib.pyplot as plt
import numpy as np
# 指定使用的那块GPU
os.environ['CUDA_VISIBLE
原创
2021-04-13 18:50:54 ·
478 阅读 ·
3 评论