def save_default_full_model_metadata(self, default_model_weights,
num_classes, filename_model_out,
filename_metadata_out):
out_model = detector_YOLO3_core.yolo_body_full(
Input(shape=(None, None, 3)), 3, num_classes)
default_model = load_model(default_model_weights, compile=False)
detector_YOLO3_core.assign_weights(default_model, out_model)
out_model.save(filename_model_out)
input_image_size, class_names, anchors, anchor_mask, obj_threshold, nms_threshold = tools_YOLO.init_default_metadata_full(
num_classes)
tools_YOLO.save_metadata(filename_metadata_out, input_image_size,
class_names, anchors, anchor_mask,
obj_threshold, nms_threshold)
return
def init_tiny(self, default_weights_h5, num_classes):
self.input_image_size, self.class_names, self.anchors, self.anchor_mask, self.obj_threshold, self.nms_threshold = tools_YOLO.init_default_metadata_tiny(
num_classes)
self.colors = tools_YOLO.generate_colors(len(self.class_names))
self.input_tensor_shape = K.placeholder(shape=(2, ))
self.sess = K.get_session()
self.model = detector_YOLO3_core.yolo_body_tiny(
Input(shape=(None, None, 3)), 3, num_classes)
default_model = load_model(default_weights_h5, compile=False)
detector_YOLO3_core.assign_weights(default_model, self.model)
self.boxes, self.scores, self.classes = detector_YOLO3_core.get_tensors_box_score_class(
self.model.output,
self.anchors,
self.anchor_mask,
len(self.class_names),
self.input_tensor_shape,
score_threshold=self.obj_threshold,
iou_threshold=self.nms_threshold)
return