constants.py

# shape of input image to YOLO
W, H =  448, 448
# grid size after last convolutional layer of YOLO
S = 7 
# anchors of YOLO model
ANCHORS = [[1.5340836003942058, 1.258424277571925],
 [1.4957766780406023, 2.2319885681948217],
 [1.2508985343739407, 0.8233350471152914]]
# number of anchors boxes
BOX = len(ANCHORS)
# maximum number of stamps on image
STAMP_NB_MAX = 10
# minimal confidence of presence a stamp in the grid cell
OUTPUT_THRESH = 0.7
# maximal iou score to consider boxes different
IOU_THRESH = 0.3
# path to folder containing images
IMAGE_FOLDER = './data/images'
# path to .cvs file containing annotations
ANNOTATIONS_PATH = './data/all_annotations.csv'
# standard deviation and mean of pixel values for normalization
STD = (0.229, 0.224, 0.225)
MEAN = (0.485, 0.456, 0.406)
# box color to show the bounding box on image
BOX_COLOR = (0, 0, 255)


# dimenstion of image embedding 
Z_DIM = 128
# hidden dimensions for encoder model
ENC_HIDDEN_DIM = 16
# hidden dimensions for decoder model
DEC_HIDDEN_DIM = 64