readme.md

Swin Transformer for Video

Model Architecture

• Type: Video Swin Transformer (Tiny - Swin3D-T)
• Source: Torchvision models.swin3d_t.
• Modifications: Classification head (Linear) modified to output 2 classes (Binary).
• Features: Hierarchical transformer with shifted windows, adapted for 3D Video processing.

Dataset Structure

Expects Dataset folder in parent directory.

Dataset/
├── violence/
└── no-violence/

How to Run

1. Install dependencies: torch, opencv-python, scikit-learn, numpy, torchvision.
2. Run python train.py.