Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/akozd/tensorflow_yolo_v3
/ functions
Functions
25 in github.com/akozd/tensorflow_yolo_v3
⨍
Functions
25
◇
Types & classes
0
↓ 75 callers
Function
convolutional_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 callers
Function
shortcut_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 callers
Function
rand
Returns a random number between a and b.
utils/utils.py:228
↓ 3 callers
Function
load_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 callers
Function
loss_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 callers
Function
yolo_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 callers
Function
get_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 callers
Function
get_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 callers
Function
get_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 callers
Function
route_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 callers
Function
upsample_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 callers
Function
yolo_v3
The fully defined architecture of the version three yolo model. Parameters ---------- inputs : tensor An array of shape:
models/yolo_v3.py:8
↓ 1 callers
Function
_main
()
train.py:16
↓ 1 callers
Function
_main
()
detect.py:9
↓ 1 callers
Function
augment_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 callers
Function
convert_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 callers
Function
create_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 callers
Function
draw_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 callers
Function
iou
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 callers
Function
iou
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 callers
Function
non_max_suppression
Applies non-max suppression to predicted boxes. Parameters ---------- predictions_with_boxes : ndarray An array of shape:
utils/utils.py:55
↓ 1 callers
Function
prepare_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 callers
Function
process_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 callers
Function
yolo_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
Function
str2bool
(v)
train.py:8