文件名称:objects_counting_dmap:从密度图计数的对象
文件大小:584KB
文件格式:ZIP
更新时间:2024-06-18 23:57:54
Python
通过使用卷积神经网络估计密度图来计数的对象 的 PyTorch 实现。 有关该方法的更多详细信息,请参阅。 数据准备 考虑了三个数据集: cell : mall : ucsd : 可以使用get_data.py脚本获取它们: Usage: get_data.py [OPTIONS] Get chosen dataset and generate HDF5 files with training and validation samples. Options: --dataset [cell|mall|ucsd] [required] --help Show this message and exit. 该脚本下载原始数据并对其进行预处理以创建带有图像和标签(对应的密度图)的 HDF5 文件。 训练 目前有两种网络架构可用:
【文件预览】:
objects_counting_dmap-master
----.gitignore(33B)
----infer.py(4KB)
----requirements.txt(138B)
----looper.py(5KB)
----LICENSE(11KB)
----examples()
--------example_cell.png(93KB)
--------example_mall.png(428KB)
--------example_ucsd.png(45KB)
----README.md(5KB)
----data_loader.py(3KB)
----get_data.py(10KB)
----model.py(7KB)
----train.py(5KB)