erprogs / GenConViT

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

Update pred_func.py #6

Closed nguyenvulong closed 9 months ago

nguyenvulong commented 9 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.