admin.suherdy/vision-ai-test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update activity_analyzer.py

Browse Source
This commit is contained in:
admin.suherdy 2025-09-30 03:58:56 +00:00
parent 903b82e4db
commit 453b46c3f6
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
activity_analyzer.py
View File

@ -103,10 +103,11 @@ class ActivityAnalyzer:
# Note: This is a placeholder - you would need to select an appropriate model
# For example: torch.hub.load('facebookresearch/pytorchvideo', 'slow_r50', pretrained=True)
# model = torch.hub.load('facebookresearch/pytorchvideo', 'slow_r50', pretrained=True)
model_name='X3D_L'
model_name='slowfast_r50'
model_dir='C:/Users/suherdy.yacob/.cache/torch/hub/checkpoints/'
#model = torch.hub.load(repo_or_dir=model_dir,model = model_name, pretrained=True)
#model = torch.hub.load(model_dir, model_name, source='local', pretrained=True)
# 'slow_r50', 'x3d_m', x3d_l', 'slowfast_r50', 'slowfast_r101', 'c2d', 'i2d', 'mvit'
model = torch.hub.load('facebookresearch/pytorchvideo', 'slow_r50', pretrained=True)
# Modify the final layer to match our number of activity classes
# This is a simplified approach - in practice you might need a more complex adaptation