File size: 926 Bytes
81d7389 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
# Two-Stream Violence Detection Network
## Model Architecture
- **Type**: Two-Stream Network (Spatial + Temporal)
- **Streams**:
1. **RGB Stream**: ResNet3D (r3d_18) to process raw video frames. Captures appearance info.
2. **Optical Flow Stream**: ResNet3D (r3d_18) to process computed dense optical flow. Captures motion info.
- **Fusion**: Features from both streams are concatenated and passed through fully connected layers.
- **Input**: 16 Frames (RGB) + 16 Flow Fields (Computed on the fly).
- **Computation**: Optical flow is computed using Farneback algorithm within the Dataloader.
## Dataset Structure
Expects `Dataset` folder in parent directory.
```
Dataset/
├── violence/
└── no-violence/
```
## How to Run
1. Install dependencies: `torch`, `opencv-python` (with contrib if needed for some algorithms, but Farneback is standard), `torchvision`.
2. Run `python train.py`.
|