LoveSiameseCat / FAViT

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool


This repository contains the official PyTorch implementation for "Forgery-Aware Adaptive Vision Transformer for Generalized Face Forgery Detection".


Requirements

  • timm == 0.5.4
  • pytorch == 1.8.0

Trained model preperation

The trained model (on FF++ C23) can be accessed from here https://drive.google.com/file/d/1QYuFeLDlbQW1uzzID31e5c0o6LaBbtl4/view?usp=drive_link

Useage

We provide a demo in evaluate.py, which makes it very easy to evaluate the trained model on your customized dataset.

About


Languages

Language:Python 100.0%