search

abhijitjadhav1998 / Deepfake_detection_using_deep_learning

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

Prediction issue #32

Open MohaYunus opened 2 months ago

MohaYunus commented 2 months ago

I am facing problem when I upload vedio it's is showing prediction error in it what should I do to solve the error or sometime I get assertion error IMG-20240406-WA0003 IMG-20240406-WA0002 IMG-20240406-WA0004

rishabhbs commented 2 months ago

did you find any solution for this problem?

profvsanchez commented 1 week ago

You installed Torch without CUDA support - it states that in the error message.

You need to install torch using this pip command

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html