erprogs / GenConViT

Deepfake Video Detection Using Generative Convolutional Vision Transformer
GNU General Public License v3.0
51 stars 11 forks source link

Update pred_func.py #6

Closed nguyenvulong closed 10 months ago

nguyenvulong commented 10 months ago

When the frame = 1, y_pred will only have one dimension. Please see https://github.com/erprogs/GenConViT/issues/5

This code will handle both cases when frame is 1 or larger than 1.