search

CommissarMa / CSRNet-pytorch

The simple implemention of CSRNet.
MIT License
49 stars 10 forks source link

test.py cannot work #9

Open xxsweetseason opened 3 years ago

xxsweetseason commented 3 years ago

The test.py cannot work, do someone know how to modify it? Or update a new version?

kuikuichio commented 3 years ago

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