1.Python 3.8.0, Numpy 1.23, Pytorch 1.13.1
2.Anaconda, cuda, and cudnn are recommended
3.The necessary settings for training are in the configs.py
4.The model is in the file models.py
5.Please check the data processing files in the datasets folder before formal training.
If you want to fully train the model, execute the following command：
python train.py
The command entered during the test is：
python test.py