MCPcopy Create free account

hub / github.com/akozd/tensorflow_yolo_v3 / functions

Functions25 in github.com/akozd/tensorflow_yolo_v3

↓ 75 callersFunctionconvolutional_layer
An implementation of the yolov3 custom convolutional layer. Parameters ---------- name : string The name of the tensor to b
models/layers/convolutional_layer.py:3
↓ 23 callersFunctionshortcut_layer
Creates the typical residual block architecture. Residual blocks are useful for training very deep convolutional neural networks because the
models/layers/shortcut_layer.py:3
↓ 13 callersFunctionrand
Returns a random number between a and b.
utils/utils.py:228
↓ 3 callersFunctionload_data
Takes the box_data and maps it into the y_true numpy array for a particular grid size. The yolov3 model has three grid siz
utils/utils.py:523
↓ 3 callersFunctionloss_per_scale
Calculates and returns the loss associated with a particular layer scale. Parameters ---------- name : string
models/loss_functions/yolo_v3_loss.py:64
↓ 3 callersFunctionyolo_layer
Takes the output of a convolutional layer and transforms it into fully scaled bounding box coordinates. These predicted boxes can then be pas
models/layers/yolo_layer.py:3
↓ 2 callersFunctionget_anchors
Reads the anchors from a file and returns them in a list. Parameters ---------- anchors_path : string Path that points to wh
utils/utils.py:252
↓ 2 callersFunctionget_classes
Reads the class names from a file and returns them in a list. Parameters ---------- classes_path : string Path that points t
utils/utils.py:232
↓ 2 callersFunctionget_training_batch
Takes the annotion lines, reads them, and from their information constructs the necessary numpy arrays that store the data to train the yolo
utils/utils.py:649
↓ 2 callersFunctionroute_layer
Takes the inputs tensor and route tensor and then concatenates them along the last axis (the channel axis). Parameters ----------
models/layers/route_layer.py:3
↓ 2 callersFunctionupsample_layer
Takes the outputs of the previous convolutional layer and upsamples them by a factor of two using the 'nearest neighbor' method. Paramet
models/layers/upsample_layer.py:3
↓ 2 callersFunctionyolo_v3
The fully defined architecture of the version three yolo model. Parameters ---------- inputs : tensor An array of shape:
models/yolo_v3.py:8
↓ 1 callersFunction_main
()
train.py:16
↓ 1 callersFunction_main
()
detect.py:9
↓ 1 callersFunctionaugment_data
Takes the iamge and box data and applies random transformations if the 'random' parameter is set to true. Otherwise, the image and box data w
utils/utils.py:337
↓ 1 callersFunctionconvert_box_coordinates
Converts coordinates in the form of center_x, center_y, width, height to min_x, min_y, max_x, max_y. The coordinate values are already scale
utils/utils.py:157
↓ 1 callersFunctioncreate_y_true
A wrapper function for creating the full y_true and y_true_box_data numpy arrays used for training the yolov3 model. Pa
utils/utils.py:479
↓ 1 callersFunctiondraw_boxes
Draws identified boxes along with class probabilities on the original image and then saves the image with the output file name. Paramete
utils/utils.py:4
↓ 1 callersFunctioniou
Calculates the intersection over union (IOU) of two bounding boxes, which is the ratio of the area where the two boxes overlap to th
utils/utils.py:84
↓ 1 callersFunctioniou
Calculates the IOU (Intersection over Union) between the predicted boxes (prct_yolo_outputs) and the true boxes (y_true_boxes
models/loss_functions/yolo_v3_loss.py:141
↓ 1 callersFunctionnon_max_suppression
Applies non-max suppression to predicted boxes. Parameters ---------- predictions_with_boxes : ndarray An array of shape:
utils/utils.py:55
↓ 1 callersFunctionprepare_data
Takes the raw data from the text file and splits it up into a training set and a validation set based on the train/val split hyperparameter.
utils/utils.py:274
↓ 1 callersFunctionprocess_image
Takes any image and transforms it into the format needed for object detection with yolov3. Parameters ---------- image_path : string
utils/utils.py:195
↓ 1 callersFunctionyolo_v3_loss
A wrapper function that returns the loss associated with a forward pass of the yolo_v3 model. The main purpose of this function is to extract
models/loss_functions/yolo_v3_loss.py:3
Functionstr2bool
(v)
train.py:8