RaphaelMeudec

from pathlib import Path

import tensorflow as tf

def select_patch(sharp, blur, patch_size_x, patch_size_y):
    """
    Select a patch on both sharp and blur images at the same localization.

    Args:
        sharp (tf.Tensor): Tensor for the sharp image

RaphaelMeudec

from pathlib import Path

import tensorflow as tf

def select_patch(sharp, blur, patch_size_x, patch_size_y):
    """
    Select a patch on both sharp and blur images at the same localization.
    Args:
        sharp (tf.Tensor): Tensor for the sharp image
        blur (tf.Tensor): Tensor for the blur image

RaphaelMeudec

# Define multi-gpu strategy 
mirrored_strategy = tf.distribute.MirroredStrategy()
# Update batch size value
batch_size *= mirrored_strategy.num_replicas_in_sync
# Create strategy scope to perform training
with mirrored_strategy.scope():
    model = [...]
    model.fit(...)

RaphaelMeudec

policy = tf.keras.mixed_precision.experimental.Policy('mixed_float16')
tf.keras.mixed_precision.experimental.set_policy(policy) 

RaphaelMeudec

"""
Performance comparison between ImageDataGenerator and tf.data.Dataset on a simple task

Dependencies available at https://www.github.com/sicara/tf2-evonorm
"""
import math
from pathlib import Path

import click
import tensorflow as tf

RaphaelMeudec

#!/usr/bin/python3
from datetime import datetime

import psutil


DEFAULT_DRAGO_NICENESS_VALUE = 5
DURATION_BEFORE_RENICING = 30