torch >= 1.7.0
yaml
omegaconf
visdom
torchmetrics
Pillow 8.2.0
-
if you have opened the visualization, you should start the visdom first.
call start_visdom.batYou can see the training lines in localhost:8097(Default)
-
Then start the demo directly
python demo.py
- The code is provided by Hui Zeng from NUDT. If you have any question, please contact zenghui35@outlook.com.