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…)