There are the overall of toolkit for Deep Learning.
https://github.com/Gojay001/toolkit-DeepLearning
Contents
. |
DataProcess
DataProcess/
data_aug.py
: augment data with transforms to aug file.
DataProcess/data_config.py
: config (hyper-)parameters.
DataProcess/data_loader.py
: load data to tensor of DataSet type.
DataProcess/divide_data.py
: divide data to train and valid files.
DataProcess/img2video.py
: transform images set to video using opencv.
DataProcess/img_resize.py
: resize images to specific size using opencv.
DataProcess/video_extract.py
: extract each frame of video to images file.
utils
utils/
bbox_iou.py
: calculate iou between two bounding-box.
utils/nms/nms_cpu.py
: remove useless bounding-box by nms(Non-maximum suppression).
(Updating…)