Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/allanzelener/YAD2K
/ functions
Functions
43 in github.com/allanzelener/YAD2K
⨍
Functions
43
◇
Types & classes
0
↓ 11 callers
Function
DarknetConv2D_BN_Leaky
Darknet Convolution2D followed by BatchNormalization and LeakyReLU.
yad2k/models/keras_darknet19.py:25
↓ 7 callers
Function
get_ids
Get image identifiers for corresponding list of dataset identifies. Parameters ---------- voc_path : str Path to VOCdevkit direct
voc_conversion_scripts/voc_to_hdf5.py:103
↓ 5 callers
Function
compose
Compose arbitrarily many functions, evaluated left to right. Reference: https://mathieularose.com/function-composition-in-python/
yad2k/utils/utils.py:6
↓ 5 callers
Function
yolo_head
Convert final layer features to bounding box parameters. Parameters ---------- feats : tensor Final convolutional layer features.
yad2k/models/keras_yolo.py:66
↓ 4 callers
Function
add_to_dataset
Process all given ids and adds them to given datasets.
voc_conversion_scripts/voc_to_hdf5.py:127
↓ 3 callers
Function
DarknetConv2D
Wrapper to set Darknet weight regularizer for Convolution2D.
yad2k/models/keras_darknet19.py:18
↓ 3 callers
Function
bottleneck_block
Bottleneck block of 3x3, 1x1, 3x3 convolutions.
yad2k/models/keras_darknet19.py:35
↓ 3 callers
Function
get_anno_path
Get path to image annotation for given year and image id.
voc_conversion_scripts/voc_to_tfrecords.py:143
↓ 3 callers
Function
get_image_path
Get path to image for given year and image id.
voc_conversion_scripts/voc_to_tfrecords.py:137
↓ 3 callers
Function
yolo_body
Create YOLO_V2 model CNN body in Keras.
yad2k/models/keras_yolo.py:45
↓ 3 callers
Function
yolo_eval
Evaluate YOLO model on given input batch and return filtered boxes.
yad2k/models/keras_yolo.py:323
↓ 2 callers
Function
bottleneck_x2_block
Bottleneck block of 3x3, 1x1, 3x3, 1x1, 3x3 convolutions.
yad2k/models/keras_darknet19.py:43
↓ 2 callers
Function
create_model
returns the body of the model and the model # Params: load_pretrained: whether or not to load the pretrained model or initialize all we
retrain_yolo.py:162
↓ 2 callers
Function
darknet_body
Generate first 18 conv layers of Darknet-19.
yad2k/models/keras_darknet19.py:51
↓ 2 callers
Function
draw_boxes
Draw bounding boxes on image. Draw bounding boxes with class name and optional box score on image. Args: image: An `array` of shape
yad2k/utils/draw_boxes.py:29
↓ 2 callers
Function
preprocess_true_boxes
Find detector in YOLO where ground truth box should appear. Parameters ---------- true_boxes : array List of ground truth boxes i
yad2k/models/keras_yolo.py:352
↓ 2 callers
Function
process_dataset
Process selected Pascal VOC dataset to generate TFRecords files. Parameters ---------- name : string Name of resulting dataset 't
voc_conversion_scripts/voc_to_tfrecords.py:149
↓ 1 callers
Function
_main
(args)
retrain_yolo.py:48
↓ 1 callers
Function
_main
(args)
yad2k.py:66
↓ 1 callers
Function
_main
(args)
test_yolo.py:56
↓ 1 callers
Function
_main
(args)
train_overfit.py:50
↓ 1 callers
Function
_main
Locate files for train and test sets and then generate TFRecords.
voc_conversion_scripts/voc_to_tfrecords.py:204
↓ 1 callers
Function
_main
(args)
voc_conversion_scripts/voc_to_hdf5.py:137
↓ 1 callers
Function
convert_to_example
Convert Pascal VOC ground truth to TFExample protobuf. Parameters ---------- image_data : bytes Encoded image bytes. boxes :
voc_conversion_scripts/voc_to_tfrecords.py:85
↓ 1 callers
Function
draw
Draw bounding boxes on image data
retrain_yolo.py:287
↓ 1 callers
Function
get_anchors
loads the anchors from a file
retrain_yolo.py:94
↓ 1 callers
Function
get_boxes_for_id
Get object bounding boxes annotations for given image. Parameters ---------- voc_path : str Path to VOCdevkit directory. year
voc_conversion_scripts/voc_to_hdf5.py:39
↓ 1 callers
Function
get_classes
loads the classes
retrain_yolo.py:87
↓ 1 callers
Function
get_colors_for_classes
Return list of random colors for number of classes given.
yad2k/utils/draw_boxes.py:10
↓ 1 callers
Function
get_detector_mask
Precompute detectors_mask and matching_true_boxes for training. Detectors mask is 1 for each spatial position in the final conv layer and
retrain_yolo.py:147
↓ 1 callers
Function
get_image_for_id
Get image data as uint8 array for given image. Parameters ---------- voc_path : str Path to VOCdevkit directory. year : str
voc_conversion_scripts/voc_to_hdf5.py:78
↓ 1 callers
Function
process_anno
Process Pascal VOC annotations.
voc_conversion_scripts/voc_to_tfrecords.py:58
↓ 1 callers
Function
process_data
processes the data
retrain_yolo.py:105
↓ 1 callers
Function
process_image
Decode image at given path.
voc_conversion_scripts/voc_to_tfrecords.py:45
↓ 1 callers
Function
train
retrain/fine-tune the model logs training with tensorboard saves training weights in current directory best weights according to v
retrain_yolo.py:230
↓ 1 callers
Function
unique_config_sections
Convert all config sections to have unique names. Adds unique suffixes to config sections for compability with configparser.
yad2k.py:46
↓ 1 callers
Function
yolo_boxes_to_corners
Convert YOLO box predictions to bounding box corners.
yad2k/models/keras_yolo.py:139
↓ 1 callers
Function
yolo_filter_boxes
Filter YOLO boxes based on object and class confidence.
yad2k/models/keras_yolo.py:309
Function
darknet19
Generate Darknet-19 model for Imagenet classification.
yad2k/models/keras_darknet19.py:67
Function
space_to_depth_x2
Thin wrapper for Tensorflow space_to_depth with block_size=2.
yad2k/models/keras_yolo.py:27
Function
space_to_depth_x2_output_shape
Determine space_to_depth output shape for block_size=2. Note: For Lambda with TensorFlow backend, output shape may not be needed.
yad2k/models/keras_yolo.py:35
Function
yolo
Generate a complete YOLO_v2 localization model.
yad2k/models/keras_yolo.py:301
Function
yolo_loss
YOLO localization loss function. Parameters ---------- yolo_output : tensor Final convolutional layer features. true_boxes :
yad2k/models/keras_yolo.py:152