update 52

Author: dragen1860

深度学习与TensorFlow入门实战-源码和PPT/lesson52-自定义数据集和迁移学习/train_scratch.py

@@ -1,5 +1,7 @@
 import  os
-os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
+
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
+
 
 import  tensorflow as tf
 import  numpy as np
@@ -8,7 +10,7 @@
 from    tensorflow.keras.callbacks import EarlyStopping
 
 tf.random.set_seed(22)
-np.random.seed(22) 
+np.random.seed(22)
 assert tf.__version__.startswith('2.')
 # 设置GPU显存按需分配
 gpus = tf.config.experimental.list_physical_devices('GPU') 

深度学习与TensorFlow入门实战-源码和PPT/lesson52-自定义数据集和迁移学习/train_transfer.py

@@ -1,6 +1,4 @@
 import  os
-os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
-
 import  tensorflow as tf
 import  numpy as np
 from    tensorflow import keras
@@ -16,7 +14,8 @@
 from    tensorflow.keras.callbacks import EarlyStopping
 
 tf.random.set_seed(22)
-np.random.seed(22) 
+np.random.seed(22)
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 assert tf.__version__.startswith('2.')
 
 
@@ -44,21 +43,20 @@ def preprocess(x,y):
 
 
 batchsz = 128
-
-# creat train db
+# 创建训练集Datset对象
 images, labels, table = load_pokemon('pokemon',mode='train')
 db_train = tf.data.Dataset.from_tensor_slices((images, labels))
 db_train = db_train.shuffle(1000).map(preprocess).batch(batchsz)
-# crate validation db
+# 创建验证集Datset对象
 images2, labels2, table = load_pokemon('pokemon',mode='val')
 db_val = tf.data.Dataset.from_tensor_slices((images2, labels2))
 db_val = db_val.map(preprocess).batch(batchsz)
-# create test db
+# 创建测试集Datset对象
 images3, labels3, table = load_pokemon('pokemon',mode='test')
 db_test = tf.data.Dataset.from_tensor_slices((images3, labels3))
 db_test = db_test.map(preprocess).batch(batchsz)
 
-
+# 
 net = keras.applications.VGG19(weights='imagenet', include_top=False,
                                pooling='max')
 net.trainable = False