File crowd_counting_error_metrics.py changed (mode: 100644) (index 91ba7e1..a6b8fb2) |
... |
... |
class CrowdCountingMeanSSIMabs(Metric): |
138 |
138 |
pad_density_map_tensor[:, 0, :y_pred.shape[2],:y_pred.shape[3]] = y_pred |
pad_density_map_tensor[:, 0, :y_pred.shape[2],:y_pred.shape[3]] = y_pred |
139 |
139 |
y_pred = pad_density_map_tensor |
y_pred = pad_density_map_tensor |
140 |
140 |
|
|
141 |
|
ssim_metric = piq.ssim(y, y_pred, reduction=sum()) |
|
|
141 |
|
ssim_metric = piq.ssim(y, y_pred, reduction="sum") |
142 |
142 |
|
|
143 |
143 |
self._sum += ssim_metric.item() |
self._sum += ssim_metric.item() |
144 |
144 |
# we multiply because ssim calculate mean of each image in batch |
# we multiply because ssim calculate mean of each image in batch |
|
... |
... |
class CrowdCountingMeanPSNRabs(Metric): |
178 |
178 |
pad_density_map_tensor[:, 0, :y_pred.shape[2], :y_pred.shape[3]] = y_pred |
pad_density_map_tensor[:, 0, :y_pred.shape[2], :y_pred.shape[3]] = y_pred |
179 |
179 |
y_pred = pad_density_map_tensor |
y_pred = pad_density_map_tensor |
180 |
180 |
|
|
181 |
|
psnr_metric = piq.psnr(y, y_pred, reduction=sum()) |
|
|
181 |
|
psnr_metric = piq.psnr(y, y_pred, reduction="sum") |
182 |
182 |
|
|
183 |
183 |
self._sum += psnr_metric.item() |
self._sum += psnr_metric.item() |
184 |
184 |
# we multiply because ssim calculate mean of each image in batch |
# we multiply because ssim calculate mean of each image in batch |
|
... |
... |
class CrowdCountingMeanSSIMclamp(Metric): |
218 |
218 |
pad_density_map_tensor[:, 0, :y_pred.shape[2], :y_pred.shape[3]] = y_pred |
pad_density_map_tensor[:, 0, :y_pred.shape[2], :y_pred.shape[3]] = y_pred |
219 |
219 |
y_pred = pad_density_map_tensor |
y_pred = pad_density_map_tensor |
220 |
220 |
|
|
221 |
|
ssim_metric = piq.ssim(y, y_pred, reduction=sum()) |
|
|
221 |
|
ssim_metric = piq.ssim(y, y_pred, reduction="sum") |
222 |
222 |
|
|
223 |
223 |
self._sum += ssim_metric.item() |
self._sum += ssim_metric.item() |
224 |
224 |
# we multiply because ssim calculate mean of each image in batch |
# we multiply because ssim calculate mean of each image in batch |
|
... |
... |
class CrowdCountingMeanPSNRclamp(Metric): |
256 |
256 |
pad_density_map_tensor[:, 0, :y_pred.shape[2], :y_pred.shape[3]] = y_pred |
pad_density_map_tensor[:, 0, :y_pred.shape[2], :y_pred.shape[3]] = y_pred |
257 |
257 |
y_pred = pad_density_map_tensor |
y_pred = pad_density_map_tensor |
258 |
258 |
|
|
259 |
|
psnr_metric = piq.psnr(y, y_pred, reduction=sum()) |
|
|
259 |
|
psnr_metric = piq.psnr(y, y_pred, reduction="sum") |
260 |
260 |
|
|
261 |
261 |
self._sum += psnr_metric.item() |
self._sum += psnr_metric.item() |
262 |
262 |
# we multiply because ssim calculate mean of each image in batch |
# we multiply because ssim calculate mean of each image in batch |