Spaces:
No application file
No application file
File size: 952 Bytes
ddec2b7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | import tensorflow as tf
import numpy as np
import os
print("=" * 60)
print("TRANSFER LEARNING TRAINING")
print("=" * 60)
print("Building MobileNetV3 with ImageNet weights...")
base_model = tf.keras.applications.MobileNetV3Small(weights="imagenet", include_top=False, input_shape=(224,224,3))
base_model.trainable = False
model = tf.keras.Sequential([
base_model,
tf.keras.layers.GlobalAveragePooling2D(),
tf.keras.layers.Dense(64, activation="relu"),
tf.keras.layers.Dropout(0.3),
tf.keras.layers.Dense(2, activation="softmax")
])
model.compile(optimizer="adam", loss="categorical_crossentropy", metrics=["accuracy"])
print("Model built!")
X = np.random.random((50, 224, 224, 3))
y = tf.keras.utils.to_categorical(np.random.randint(0, 2, 50), 2)
print("Training...")
model.fit(X, y, epochs=3, batch_size=8)
os.makedirs("data/models", exist_ok=True)
model.save("data/models/transfer_learning_model.h5")
print("Model saved!")
|