Code for the paper "Audio-Visual Localization by Acoustic Image Generation", AAAI 2021
- Python 3
- TensorFlow 1.14.0 >=
We provide several scripts and a package with all the necessary code for training and testing our model. The code is organized in several folders and a couple of main scripts as follows:
-
The
mainscript is used for training and testing the different models: UNet, DualCamNet with real and generated images. -
The
showimages_bbplots FlickrSoundnet energy from a UNet checkpoint and list of tfrecords of FlickrSoundnet. -
The
showimagesplots ACIVW and AVIA energy from a UNet checkpoint and list of testing tfrecords. -
The
showvideoplots VGG Sound energy from a UNet checkpoint and list of tfrecords of a video. -
The
meanstdcomputes the mean the computed metrics of 5 experiments excluding for each metric min and max values and saves them in a xlsx file.