├── dataLoadess.py ├── gan_resnet.py ├── model └── save_model │ ├── first_frame_net_model_epoch_100.pth │ ├── rnn1_model_epoch_100.pth │ └── rnn2_model_epoch_100.pth ├── models.py ├── my_tools.py ├── readme.md ├── real data ├── BIRNAT_domino.mat ├── BIRNAT_water_balloon.mat └── BIRNAT_wheel.mat ├── test.py ├── test ├── aerial32_cacti.mat ├── crash32_cacti.mat ├── drop8_cacti.mat ├── kobe_cacti.mat ├── runner8_cacti.mat └── traffic_cacti.mat ├── train.py ├── train ├── data_generation.m └── mask.mat ├── train_at.py └── utils.py /dataLoadess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/dataLoadess.py -------------------------------------------------------------------------------- /gan_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/gan_resnet.py -------------------------------------------------------------------------------- /model/save_model/first_frame_net_model_epoch_100.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/model/save_model/first_frame_net_model_epoch_100.pth -------------------------------------------------------------------------------- /model/save_model/rnn1_model_epoch_100.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/model/save_model/rnn1_model_epoch_100.pth -------------------------------------------------------------------------------- /model/save_model/rnn2_model_epoch_100.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/model/save_model/rnn2_model_epoch_100.pth -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/models.py -------------------------------------------------------------------------------- /my_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/my_tools.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/readme.md -------------------------------------------------------------------------------- /real data/BIRNAT_domino.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/real data/BIRNAT_domino.mat -------------------------------------------------------------------------------- /real data/BIRNAT_water_balloon.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/real data/BIRNAT_water_balloon.mat -------------------------------------------------------------------------------- /real data/BIRNAT_wheel.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/real data/BIRNAT_wheel.mat -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/test.py -------------------------------------------------------------------------------- /test/aerial32_cacti.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/test/aerial32_cacti.mat -------------------------------------------------------------------------------- /test/crash32_cacti.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/test/crash32_cacti.mat -------------------------------------------------------------------------------- /test/drop8_cacti.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/test/drop8_cacti.mat -------------------------------------------------------------------------------- /test/kobe_cacti.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/test/kobe_cacti.mat -------------------------------------------------------------------------------- /test/runner8_cacti.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/test/runner8_cacti.mat -------------------------------------------------------------------------------- /test/traffic_cacti.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/test/traffic_cacti.mat -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/train.py -------------------------------------------------------------------------------- /train/data_generation.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/train/data_generation.m -------------------------------------------------------------------------------- /train/mask.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/train/mask.mat -------------------------------------------------------------------------------- /train_at.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/train_at.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BoChenGroup/BIRNAT/HEAD/utils.py --------------------------------------------------------------------------------