Upload 32 files

DenseAV/denseav/data/AVDatasets.py

@@ -18,13 +18,9 @@ from PIL import Image
 from torch.utils.data import Dataset, DataLoader, default_collate, Subset, ConcatDataset
 from tqdm import tqdm
 
-import sys
-sys.path.append('../constants')
-sys.path.append('../shared')
-
-from constants import AUDIO_MASK, AUDIO_POS_MASK, IMAGE_MASK, IMAGE_INPUT
-from make_tarballs import untar_all
-from shared import norm, prep_waveform
+from DenseAV.denseav.constants import AUDIO_MASK, AUDIO_POS_MASK, IMAGE_MASK, IMAGE_INPUT
+from DenseAV.denseav.data.make_tarballs import untar_all
+from DenseAV.denseav.shared import norm, prep_waveform
 
 
 def sample_choice(choices, probs):