def retrieve_training_summaries():
team_uuid = team_info.retrieve_team_uuid(flask.session, flask.request)
data = flask.request.form.to_dict(flat=True)
model_uuid = data.get('model_uuid')
retrieve_scalars = (data.get('retrieve_scalars') == 'true')
retrieve_images = (data.get('retrieve_images') == 'true')
model_entity = model_trainer.retrieve_model_entity(team_uuid, model_uuid)
training_updated, training_sorted_tags, training_sorted_steps, training_summaries = model_trainer.retrieve_training_summaries(
team_uuid, model_uuid, retrieve_scalars, retrieve_images)
eval_updated, eval_sorted_tags, eval_sorted_steps, eval_summaries = model_trainer.retrieve_eval_summaries(
team_uuid, model_uuid, retrieve_scalars, retrieve_images)
sanitize(model_entity)
response = {
'model_entity': model_entity,
'training_updated': training_updated,
'training_sorted_tags': training_sorted_tags,
'training_sorted_steps': training_sorted_steps,
'training_summaries': training_summaries,
'eval_updated': eval_updated,
'eval_sorted_tags': eval_sorted_tags,
'eval_sorted_steps': eval_sorted_steps,
'eval_summaries': eval_summaries,
}
blob_storage.set_cors_policy_for_get()
return flask.jsonify(response)
def create_tflite():
team_uuid = team_info.retrieve_team_uuid(flask.session, flask.request)
data = flask.request.form.to_dict(flat=True)
model_uuid = data.get('model_uuid')
download_url = tflite_creator.create_tflite(team_uuid, model_uuid)
response = {
'download_url': download_url,
}
blob_storage.set_cors_policy_for_get()
return flask.jsonify(response)
def retrieve_video_frame_entities_with_image_urls():
team_uuid = team_info.retrieve_team_uuid(flask.session, flask.request)
data = flask.request.form.to_dict(flat=True)
video_uuid = data.get('video_uuid')
min_frame_number = int(data.get('min_frame_number'))
max_frame_number = int(data.get('max_frame_number'))
video_frame_entities = storage.retrieve_video_frame_entities_with_image_urls(
team_uuid, video_uuid, min_frame_number, max_frame_number)
sanitize(video_frame_entities)
response = {
'video_frame_entities': video_frame_entities,
}
blob_storage.set_cors_policy_for_get()
return flask.jsonify(response)
def get_dataset_zip_status():
team_uuid = team_info.retrieve_team_uuid(flask.session, flask.request)
data = flask.request.form.to_dict(flat=True)
dataset_zip_uuid = data.get('dataset_zip_uuid')
partition_count = int(data.get('partition_count'))
exists_array, download_url_array = blob_storage.get_dataset_zip_download_url(
team_uuid, dataset_zip_uuid, partition_count)
file_count_array, files_written_array = storage.retrieve_dataset_zipper_files_written(
team_uuid, dataset_zip_uuid, partition_count)
response = {
'is_ready_array': exists_array,
'download_url_array': download_url_array,
'file_count_array': file_count_array,
'files_written_array': files_written_array,
}
blob_storage.set_cors_policy_for_get()
return flask.jsonify(response)
