Training on shanghaitech dataset

[LBAMAD]

Hello,thank you for your open-source of such great work!
However,when I generated the background images of the shanghaitech dataset,I encountered the following problem:
File "/workspace/DMAD/DMAD-PDM/Train_shanghai.py", line 57, in MNADTrain
gen_bkg_(test_folder, "./bkg_/", )
File "/workspace/DMAD/DMAD-PDM/utils.py", line 88, in gen_bkg_
frame = cv2.imread(img_path).copy()
^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'copy'
How can I solve the problem?Hope for your reply,thank you!

[FlappyPeggy]

It seems that you forget to download the background template for ShanghaiTech or set a wrong loading path (should be "./bkg_/").

Please check readme file and download them at: BaiduPan (Password:dmad) or GoogleDrive

All files are located with the same path, you just need to download them to the Corresponding Folder and Unzip.

[FlappyPeggy]

It seems that you forget to download the background template for ShanghaiTech or set a wrong loading path (should be "./bkg_/").

Please check readme file and download them at: BaiduPan (Password:dmad) or GoogleDrive

All files are located with the same path, you just need to download them to the Corresponding Folder and Unzip.

If you want to re-generate background templates, please check if the test folder (test_folder) exists.