guanfuchen / objdet

实现常用的one-stage和two-stage目标检测网络 华为媒体研究院 图文Caption、OCR识别、图视文多模态理解与生成相关方向工作或实习欢迎咨询 15757172165 https://guanfuchen.github.io/media/hw_zhaopin_20220724_tiny.jpg
55 stars 15 forks source link
faster-rcnn object-detection pytorch ssd

objdet


object detection algorithms

这个仓库旨在实现常用的目标检测算法,主要参考如下:

图片来自于deep_learning_object_detection


论文资料


小目标检测


遮挡目标检测


domain目标检测


文本检测


类别不平衡目标检测


小数据集目标检测


弱监督目标检测


非极大值抑制


困难样本采样策略


回归框loss

目前常用的回归框loss有l2 loss,smooth l1 loss和IoU loss。


网络实现


非极大值抑制


困难样例学习


数据集实现


用法

可视化

visdom

# 在tmux或者另一个终端中开启可视化服务器visdom
python -m visdom.server
# 然后在浏览器中查看127.0.0.1:9097

训练

# 训练模型
python train.py

校验

# 校验模型
python validate.py

测试

# 测试模型
python test.py

TODO