File size: 305 Bytes
83357ea
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import streamlit as st
import tensorflow as tf
import numpy as np

# Setting random seed to obtain reproducible results.
tf.random.set_seed(42)

# Initialize global variables.
AUTO = tf.data.AUTOTUNE
BATCH_SIZE = 1
NUM_SAMPLES = 32
POS_ENCODE_DIMS = 16
EPOCHS = 30
H = 25
W = 25
focal = 0.6911112070083618