LEGIONM36's picture
Upload 4 files
6028065 verified
# Violence GRU Model
## Model Architecture
- **Type**: CNN-GRU Hybrid
- **Components**:
- **CNN Backbone**: 4-Layer 2D CNN to extract spatial features from each frame.
- **Recurrent Unit**: 2-Layer GRU (Gated Recurrent Unit) to model temporal dependencies.
- **Classifier**: Fully Connected Layer.
- **Input**: Video sequence of 16 frames, resized to 112x112.
- **Output**: Binary Classification (Violence vs No-Violence).
## Dataset Structure
The code expects a `Dataset` folder in the parent directory.
```
Dataset/
β”œβ”€β”€ violence/
└── no-violence/
```
## How to Run
1. Install dependencies: `torch`, `opencv-python`, `scikit-learn`, `numpy`.
2. Run `python train.py`.