文件名称：FCHD:人头检测网络 “FAST AND ACCURATE HEAD DETECTION IN CROWDED SCENES”
文件大小：1.56MB
文件格式：ZIP
更新时间：2024-06-07 10:25:38
Python
FCHD 本工程是官方工程的python3版本，重构了部分代码，并训练得到了一个效果不错的模型。 论文地址： 官方工程地址： 安装 Clone git clone https://github.com/embracesource-cv-com/FCHD.git 安装依赖 cd FCHD pip install -r requirements.txt 数据配置 在下载Brainwash数据集，caffe预训练模型，以及人头检测模型 将预训练模型vgg16_caffe.pth放到./checkpoints/pre_trained目录下 将检测模型checkpoint_best.pth放到./checkpoints目录下 将brainwash数据集放到/home/dataset/fchd_datas目录下 当然你可以自定义各个路径，然后在config.py中修改相应的配置 预测 在工程root目
【文件预览】：
FCHD-master
----trainer.py(4KB)
----.gitignore(1KB)
----requirements.txt(541B)
----data()
--------__init__.py(0B)
--------preprocess.py(2KB)
--------dataset.py(2KB)
----networks()
--------base_net.py(622B)
--------__init__.py(0B)
--------anchor_target_layer.py(3KB)
--------detector.py(3KB)
--------proposal_layer.py(2KB)
--------rpn.py(3KB)
----utils()
--------visualize.py(6KB)
--------tools.py(5KB)
--------losses.py(769B)
----examples()
--------result_3.png(609KB)
--------test_3.jpg(138KB)
--------result_2.png(375KB)
--------result_1.png(409KB)
--------test_1.jpg(46KB)
--------test_2.jpg(41KB)
----README.md(1KB)
----config.py(745B)
----predict.py(2KB)
----train.py(4KB)