pytorch>1.0
python=3.6
datasets/
ucm/
class_1/
class_2/
...
whu/
class_1/
class_2/
...
aid/
class_1/
class_2/
...
opt/
class_1/
class_2/
...
python baseline_train.py
python train.py