Bubbliiiing's repositories
classification-pytorch
这是各个主干网络分类模型的源码,可以用于训练自己的分类模型。
facenet-retinaface-pytorch
这是一个利用facenet和retinaface实现人脸识别的库,可以进行在线的人脸识别。
faster-rcnn-tf2
这是一个faster-rcnn的tensorflow2实现的库,可以利用voc数据集格式的数据进行训练。
object-detection-augmentation
这里面存放了一些目标检测算法的数据增强方法。如mosaic、mixup。
Mobilenet-SSD-Essay
这是Mobilenet-SSD的论文版,可用于训练与预测。
mobilenet-yolov4-keras
这是一个mobilenet-yolov4的库,把yolov4主干网络修改成了mobilenet,修改了Panet的卷积组成,使参数量大幅度缩小。
yolox-keras
这是一个yolox-keras的源码,可以用于训练自己的模型。
efficientdet-tf2
这是一个efficientdet-tf2(tensorflow2)的源码,可以用于训练自己的模型。
srgan-pytorch
这是一个srgan-keras的源码,可以用于训练自己的模型。
dcgan-pytorch
这是一个dcgan-pytorch的源码,可以用于训练自己的模型。
mobilenet-yolov4-tf2
这是一个mobilenet-yolov4的库,把yolov4主干网络修改成了mobilenet,修改了Panet的卷积组成,使参数量大幅度缩小。
dcgan-keras
这是一个dcgan-keras的源码,可以用于训练自己的模型。
cyclegan-pytorch
这是一个cyclegan-pytorch的源码,可以用于训练自己的模型。
srgan-keras
这是一个srgan-keras的源码,可以用于训练自己的模型。
yolov5-keras
这是一个YoloV5-keras的源码,可以用于训练自己的模型。
yolov7-keras
这是一个yolov7-keras的源码,可以用于训练自己的模型。
yolov5-v6.1-tf2
这是一个yolov5-v6.1-tf2的源码,可以用于训练自己的模型。
yolact-keras
这是一个yolact-keras的库,可用于训练自己的数据集
efficientnet-yolo3-tf2
这是一个efficientnet-yolo3-tf2的源码,将yolov3的主干特征提取网络修改成了efficientnet
yolact-tf2
这是一个yolact-tensorflow2的库,可用于训练自己的数据集
cyclegan-keras
这是一个cyclegan-keras的源码,可以用于训练自己的模型。
fcos-keras
这是一个fcos-keras的仓库,可以用于训练自己的数据集。
yolov5-v6.1-keras
这是一个yolov5-v6.1-keras的源码,可以用于训练自己的模型。
yolov7-tiny-keras
这是一个yolov7-tiny-keras的源码,可以用于训练自己的模型。