Downloads · 30 days
3
10% of all-time downloads
manhdungcr7/cuenet-rwf2000
cuenet-rwf2000 is a video classification model from manhdungcr7. Use it for the video classification task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
CUE-Net (CLIP-based UniFormerV2 Enhanced Network) là mô hình phát hiện bạo lực từ video giám sát, được huấn luyện trên bộ dữ liệu RWF-2000.
Downloads · 30 days
3
10% of all-time downloads
All-time downloads
30
Public
Repo size
4.2 GB
Likes
1
Public
Click a slice to open those files.
.pyth4.2 GB · 100%
From the Hugging Face model README
CUE-Net (CLIP-based UniFormerV2 Enhanced Network) là mô hình phát hiện bạo lực từ video giám sát, được huấn luyện trên bộ dữ liệu RWF-2000.
| Metric | Score |
|---|---|
| Accuracy | 89.50% |
| F1-Score | 89.48% |
from slowfast.models.build import build_model
from slowfast.config.defaults import get_cfg
cfg = get_cfg()
cfg.merge_from_file("config.yaml")
model = build_model(cfg)
# Load checkpoint
checkpoint = torch.load("cuenet_rwf2000_epoch51.pyth")
model.load_state_dict(checkpoint["model_state"])