This projects aims in detection of video deepfakes using deep learning techniques like RestNext and LSTM. We have achived deepfake detection by using transfer learning where the pretrained RestNext CNN is used to obtain a feature vector, further the LSTM layer is trained using the features. For more details follow the documentaion.
GNU General Public License v3.0
397
stars
140
forks
source link
AssertionError: Torch not compiled with CUDA enabled #7
in my macOs giving this error in model = Model(2).cuda()