CSRNet-pytorch
CSRNet-pytorch copied to clipboard
test.py cannot work
The test.py cannot work, do someone know how to modify it? Or update a new version?
Please run this code in the test.py file dataset = CrowdDataset(img_root, gt_dmap_root,8), how to solve the problem of int8? Assign a value of 8 to main_transform, then execute to img, dmap = self.main_transform((img, dmap)), the data type does not match,Thank you