Crowding Counter 之根据.mat标注制作密度图进行训练

通常，corwding counter任务中数据集只有图像和对应的标注文件，标注文件中为每个人的坐标（一个人对应一个坐标）。如果需要采用density map进行训练的话，需要生成对应的density map。

本代码用于将mat文件生成对应的density map；最终用.h5文件保存；

import  h5py
import  scipy.io as io
import PIL.Image as Image
import numpy as np
import os
import glob
from matplotlib import pyplot as plt
from scipy.ndimage.filters import gaussian_filter
from matplotlib import cm as CM
from image import *# root is the path to ShanghaiTech dataset
root=''part_B_train = os.path.join(root,'justPoint','images')
# part_B_test = os.path.join(root,'justPoint','images')
path_sets = [part_B_train]img_paths = []
for path in path_sets:for img_path in glob.glob(os.path.join(path, '*.jpg')):img_paths.append(img_path)for img_path in img_paths:print(img_path)mat = io.loadmat(img_path.replace('.jpg','.mat').replace('images','ground_truth').replace('IMG_','GT_IMG_'))img= plt.imread(img_path)k = np.zeros((img.shape[0],img.shape[1]))gt = mat["image_info"][0,0][0,0][0]for i in range(0,len(gt)):if int(gt[i][1])<img.shape[0] and int(gt[i][0])<img.shape[1]:k[int(gt[i][1]),int(gt[i][0])]=1k = gaussian_filter(k,15)with h5py.File(img_path.replace('.jpg','.h5').replace('images','ground_truth'), 'w') as hf:hf['density'] = k

Crowding Counter 之根据.mat标注制作密度图进行训练相关推荐

Crowding Counter 之裁剪图像进行数据集扩充
现有的Crowding Counter任务数据集大小均不大,例如ShanghaiTech数据集的大小就几百张,因此为了增强训练的效果,通常会将数据集进行扩充,较为简单的方法就是将原始的图像进行局部裁剪 ...
Crowding Counter 之可视化h5文件
在进行训练前,会根据.mat标注文件形成h5文件,在此对于h5文件进行可视化: 代码比较简单,仅供学习~ import numpy as np import matplotlib.pyplot as ...
人群计数之生成密度图
一.概述在近几年人群计数领域的研究中,大多数论文都沿用了MCNN( Multi-column Convolutional Neural Network )中提出的生成密度图的方法,本文将通过代码和C ...
人群计数经典方法Density Map Estimation,密度图估计
(3)Density Map Estimation(主流) 这是crowd counting的主流方法传统方法不好在哪里?object detection-based method和regressi ...
Py之seaborn：数据可视化seaborn库(二)的组合图可视化之密度图/核密度图分布可视化、箱型图/散点图、小提琴图/散点图组合可视化的简介、使用方法之最强攻略(建议收藏)
Py之seaborn:数据可视化seaborn库(二)的组合图可视化之密度图/核密度图分布可视化.箱型图/散点图.小提琴图/散点图组合可视化的简介.使用方法之最强攻略(建议收藏) 目录二.组合图可视 ...
python区域增长_Python – 有效地为高密度区域创建密度图,稀疏区域的点
我需要制作一个剧情,就像剧情中高密度地区的密度图一样,但低于某个阈值则使用个别点.我找不到任何与matplotlib缩略图库或谷歌搜索中所需的代码类似的代码.我有一个我自己写的工作代码,但是当点/仓数 ...
seaborn分布数据可视化：直方图|密度图|散点图
系统自带的数据表格(存放在github上https://github.com/mwaskom/seaborn-data),使用时通过sns.load_dataset('表名称')即可,结果为一个Dat ...
密度图的密度估计_不同类型的二维密度图小教程
R相关小教程链接: 用R构建气泡图案例小教程 [小教程]散点图.饼图怎么在我的文章中完美展示小教程热图在论文发表中完美呈现小教程 R与密度.函数.变量的微妙关系北京市计算中心医用数据库建设解决方案 ...
graphics | 基础绘图系统（九）——栅格图、点密度图、等高线（填充）图、三维图...
先插播一则消息:刚建的交流群莫名其妙地被判违规,不仅无法加新人,而且群内的消息可能只能自己看到.先看看过几天能不能解封,不行再另想办法. 本篇是"基础绘图系统"系列的第九篇推文,也 ...

Crowding Counter 之根据.mat标注制作密度图进行训练

Crowding Counter 之根据.mat标注制作密度图进行训练相关推荐

最新文章

热门文章

Crowding Counter 之 根据.mat标注制作密度图进行训练

Crowding Counter 之 根据.mat标注制作密度图进行训练相关推荐

最新文章

热门文章

Crowding Counter 之根据.mat标注制作密度图进行训练

Crowding Counter 之根据.mat标注制作密度图进行训练相关推荐