Training settings
Please provide a valid training processor option
Neural network architecture
sys.path.append('./resources/libraries')
import os
import tensorflow as tf
from tensorflow.keras.optimizers import Adam
from tensorflow.keras.applications import MobileNetV2
from tensorflow.keras.layers import BatchNormalization, Conv2D, Softmax, Reshape
from tensorflow.keras.models import Model
from ei_tensorflow.constrained_object_detection import models, dataset, metrics, util
import ei_tensorflow.training
from pathlib import Path
import requests
def build_model(input_shape: tuple, weights: str, alpha: float,
num_classes: int) -> tf.keras.Model:
""" Construct a constrained object detection model.
Args:
input_shape: Passed to MobileNet construction.
weights: Weights for initialization of MobileNet where None implies
random initialization.
alpha: MobileNet alpha value.
num_classes: Number of classes, i.e. final dimension size, in output.
Returns:
Uncompiled keras model.
Model takes (B, H, W, C) input and
returns (B, H//8, W//8, num_classes) logits.
"""
#! First create full mobile_net_V2 from (HW, HW, C) input
#! to (HW/8, HW/8, C) output
mobile_net_v2 = MobileNetV2(input_shape=input_shape,
weights=weights,
alpha=alpha,
include_top=True)
#! Default batch norm is configured for huge networks, let's speed it up
for layer in mobile_net_v2.layers:
if type(layer) == BatchNormalization:
layer.momentum = 0.9
#! Cut MobileNet where it hits 1/8th input resolution; i.e. (HW/8, HW/8, C)
cut_point = mobile_net_v2.get_layer('block_6_expand_relu')
#! Now attach a small additional head on the MobileNet
model = Conv2D(filters=32, kernel_size=1, strides=1,
activation='relu', name='head')(cut_point.output)
logits = Conv2D(filters=num_classes, kernel_size=1, strides=1,
activation=None, name='logits')(model)
return Model(inputs=mobile_net_v2.input, outputs=logits)
def train(num_classes: int, learning_rate: float, num_epochs: int,
alpha: float, object_weight: float,
train_dataset: tf.data.Dataset,
validation_dataset: tf.data.Dataset,
best_model_path: str,
input_shape: tuple,
lr_finder: bool = False) -> tf.keras.Model:
""" Construct and train a constrained object detection model.
Args:
num_classes: Number of classes in datasets. This does not include
implied background class introduced by segmentation map dataset
conversion.
learning_rate: Learning rate for Adam.
num_epochs: Number of epochs passed to model.fit
alpha: Alpha used to construct MobileNet. Pretrained weights will be
used if there is a matching set.
object_weight: The weighting to give the object in the loss function
where background has an implied weight of 1.0.
train_dataset: Training dataset of (x, (bbox, one_hot_y))
validation_dataset: Validation dataset of (x, (bbox, one_hot_y))
best_model_path: location to save best model path. note: weights
will be restored from this path based on best val_f1 score.
input_shape: The shape of the model's input
max_training_time_s: Max training time (will exit if est. training time is over the limit)
is_enterprise_project: Determines what message we print if training time exceeds
Returns:
Trained keras model.
Constructs a new constrained object detection model with num_classes+1
outputs (denoting the classes with an implied background class of 0).
Both training and validation datasets are adapted from
(x, (bbox, one_hot_y)) to (x, segmentation_map). Model is trained with a
custom weighted cross entropy function.
"""
nonlocal callbacks
num_classes_with_background = num_classes + 1
input_width_height = None
width, height, input_num_channels = input_shape
if width != height:
raise Exception(f"Only square inputs are supported; not {input_shape}")
input_width_height = width
#! Use pretrained weights, if we have them for configured
weights = None
if input_num_channels == 1:
if alpha == 0.1:
weights = "./transfer-learning-weights/edgeimpulse/MobileNetV2.0_1.96x96.grayscale.bsize_64.lr_0_05.epoch_441.val_loss_4.13.val_accuracy_0.2.hdf5"
elif alpha == 0.35:
weights = "./transfer-learning-weights/edgeimpulse/MobileNetV2.0_35.96x96.grayscale.bsize_64.lr_0_005.epoch_260.val_loss_3.10.val_accuracy_0.35.hdf5"
elif input_num_channels == 3:
if alpha == 0.1:
weights = "./transfer-learning-weights/edgeimpulse/MobileNetV2.0_1.96x96.color.bsize_64.lr_0_05.epoch_498.val_loss_3.85.hdf5"
elif alpha == 0.35:
weights = "./transfer-learning-weights/keras/mobilenet_v2_weights_tf_dim_ordering_tf_kernels_0.35_96.h5"
if (weights and not os.path.exists(weights)):
print(f"Pretrained weights {weights} unavailable; downloading...")
p = Path(weights)
if not p.exists():
if not p.parent.exists():
p.parent.mkdir(parents=True)
root_url = 'https://cdn.edgeimpulse.com/'
weights_data = requests.get(root_url + weights[2:]).content
with open(weights, 'wb') as f:
f.write(weights_data)
print(f"Pretrained weights {weights} unavailable; downloading OK\n")
model = build_model(
input_shape=input_shape,
weights=weights,
alpha=alpha,
num_classes=num_classes_with_background
)
#! Derive output size from model
model_output_shape = model.layers[-1].output.shape
_batch, width, height, num_classes = model_output_shape
if width != height:
raise Exception(f"Only square outputs are supported; not {model_output_shape}")
output_width_height = width
#! Build weighted cross entropy loss specific to this model size
weighted_xent = models.construct_weighted_xent_fn(model.output.shape, object_weight)
#! Transform bounding box labels into segmentation maps
train_segmentation_dataset = train_dataset.map(dataset.bbox_to_segmentation(
output_width_height, num_classes_with_background)).batch(32, drop_remainder=False).prefetch(1)
validation_segmentation_dataset = validation_dataset.map(dataset.bbox_to_segmentation(
output_width_height, num_classes_with_background, validation=True)).batch(32, drop_remainder=False).prefetch(1)
#! Initialise bias of final classifier based on training data prior.
util.set_classifier_biases_from_dataset(
model, train_segmentation_dataset)
if lr_finder:
learning_rate = ei_tensorflow.lr_finder.find_lr(model, train_segmentation_dataset, weighted_xent)
model.compile(loss=weighted_xent,
optimizer=Adam(learning_rate=learning_rate))
#! Create callback that will do centroid scoring on end of epoch against
#! validation data. Include a callback to show % progress in slow cases.
callbacks = callbacks if callbacks else []
callbacks.append(metrics.CentroidScoring(validation_segmentation_dataset,
output_width_height, num_classes_with_background))
callbacks.append(metrics.PrintPercentageTrained(num_epochs))
#! Include a callback for model checkpointing based on the best validation f1.
callbacks.append(
tf.keras.callbacks.ModelCheckpoint(best_model_path,
monitor='val_f1', save_best_only=True, mode='max',
save_weights_only=True, verbose=0))
model.fit(train_segmentation_dataset,
validation_data=validation_segmentation_dataset,
epochs=num_epochs, callbacks=callbacks, verbose=0)
#! Restore best weights.
model.load_weights(best_model_path)
#! Add explicit softmax layer before export.
softmax_layer = Softmax()(model.layers[-1].output)
model = Model(model.input, softmax_layer)
return model
EPOCHS = args.epochs or 60
LEARNING_RATE = args.learning_rate or 0.001
model = train(num_classes=classes,
learning_rate=LEARNING_RATE,
num_epochs=EPOCHS,
alpha=0.35,
object_weight=100,
train_dataset=train_dataset,
validation_dataset=validation_dataset,
best_model_path=BEST_MODEL_PATH,
input_shape=MODEL_INPUT_SHAPE,
lr_finder=False)
override_mode = 'segmentation'
disable_per_channel_quantization = False
Input layer (27,648 features)
FOMO (Faster Objects, More Objects) MobileNetV2 0.35
Output layer (1 classes)
Model
Model version: