文件名称:retinaface:https的翻拍
文件大小:39KB
文件格式:ZIP
更新时间:2024-04-08 21:20:02
pytorch face-detection deeplearning object-detection retinaface
视网膜面
此仓库基于构建
差异性
火车环已移至
IT增加了一组功能:
分布式培训
fp16
同步批次标准
支持各种记录器,例如或
超参数在配置文件中定义
分散在代码中的超参数移到了
增强=>
手动实现的颜色已由Albumentations库替换。
去做:
在相册中未实现水平翻转
尚未实现诸如旋转或转置之类的空间变换。
颜色转换在配置中定义。
添加了mAP计算以进行验证
为了跟踪进度,在验证时会计算mAP指标。
安装
pip install -U retinaface_pytorch
示例推论
import cv2
from retinaface . pre_trained_models import get_model
image =
【文件预览】:
retinaface-master
----.github()
--------FUNDING.yml(678B)
--------workflows()
----.pylintrc(18KB)
----requirements.txt(61B)
----retinaface()
--------inference.py(11KB)
--------predict_single.py(5KB)
--------train.py(9KB)
--------box_utils.py(10KB)
--------configs()
--------multibox_loss.py(6KB)
--------pre_trained_models.py(674B)
--------utils.py(803B)
--------dataset.py(5KB)
--------__init__.py(22B)
--------net.py(3KB)
--------network.py(4KB)
--------prior_box.py(905B)
--------data_augment.py(5KB)
----.isort.cfg(182B)
----LICENSE(1KB)
----setup.cfg(240B)
----setup.py(3KB)
----README.md(6KB)
----.deepsource.toml(187B)
----.pre-commit-config.yaml(1KB)
----.gitignore(1KB)
----pyproject.toml(516B)