我们将使用ShangaiTech数据集。数据集分为两部分。对于本文,我们将仅与B部分一起为人群训练模型,然后在自定义数据集上对其进行测试。您可以选择使用其中任何一个部分。该代码可以与任何一个正常工作