Toolkit for DL

Toolkit for DL

There are the overall of toolkit for Deep Learning.
https://github.com/Gojay001/toolkit-DeepLearning

Contents

.
|- DataProcess/
|- data_aug.py
|- data_config.py
|- data_loader.py
|- divide_data.py
|- img2video.py
|- img_resize.py
|- video_extract.py
|- utils/
|- bbox_iou.py
|- nms
|- nms_cpu.py
|- ...
|- test.py
|- train.py

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


Comments

Your browser is out-of-date!

Update your browser to view this website correctly. Update my browser now

×