|
|
|
|
|
"""
|
|
|
Test script for GenVidBench app
|
|
|
"""
|
|
|
|
|
|
import os
|
|
|
import sys
|
|
|
|
|
|
def test_imports():
|
|
|
"""Test if all required modules can be imported"""
|
|
|
print("🧪 Testing imports...")
|
|
|
|
|
|
try:
|
|
|
import torch
|
|
|
print("✅ PyTorch imported successfully")
|
|
|
except ImportError as e:
|
|
|
print(f"❌ PyTorch import failed: {e}")
|
|
|
return False
|
|
|
|
|
|
try:
|
|
|
import gradio as gr
|
|
|
print("✅ Gradio imported successfully")
|
|
|
except ImportError as e:
|
|
|
print(f"❌ Gradio import failed: {e}")
|
|
|
return False
|
|
|
|
|
|
try:
|
|
|
from mmaction.apis import init_recognizer, inference_recognizer
|
|
|
print("✅ MMAction2 APIs imported successfully")
|
|
|
except ImportError as e:
|
|
|
print(f"❌ MMAction2 import failed: {e}")
|
|
|
return False
|
|
|
|
|
|
return True
|
|
|
|
|
|
def test_files():
|
|
|
"""Test if required files exist"""
|
|
|
print("🧪 Testing file structure...")
|
|
|
|
|
|
required_files = [
|
|
|
'app.py',
|
|
|
'requirements.txt',
|
|
|
'README.md',
|
|
|
'demo/demo_configs/tsn_r50_1x1x8_video_infer.py'
|
|
|
]
|
|
|
|
|
|
for file_path in required_files:
|
|
|
if os.path.exists(file_path):
|
|
|
print(f"✅ {file_path} exists")
|
|
|
else:
|
|
|
print(f"❌ {file_path} missing")
|
|
|
return False
|
|
|
|
|
|
return True
|
|
|
|
|
|
def test_model_checkpoint():
|
|
|
"""Test if model checkpoint exists or can be downloaded"""
|
|
|
print("🧪 Testing model checkpoint...")
|
|
|
|
|
|
checkpoint_path = 'checkpoints/tsn_r50_8xb32-1x1x8-100e_kinetics400-rgb_20220818-2692d16c.pth'
|
|
|
|
|
|
if os.path.exists(checkpoint_path):
|
|
|
print(f"✅ Model checkpoint found: {checkpoint_path}")
|
|
|
return True
|
|
|
else:
|
|
|
print(f"⚠️ Model checkpoint not found: {checkpoint_path}")
|
|
|
print(" Run 'python download_model.py' to download it")
|
|
|
return False
|
|
|
|
|
|
def main():
|
|
|
"""Run all tests"""
|
|
|
print("🚀 Testing GenVidBench setup...")
|
|
|
print("=" * 50)
|
|
|
|
|
|
tests = [
|
|
|
("Import Test", test_imports),
|
|
|
("File Structure Test", test_files),
|
|
|
("Model Checkpoint Test", test_model_checkpoint)
|
|
|
]
|
|
|
|
|
|
results = []
|
|
|
for test_name, test_func in tests:
|
|
|
print(f"\n📋 {test_name}")
|
|
|
print("-" * 30)
|
|
|
result = test_func()
|
|
|
results.append((test_name, result))
|
|
|
|
|
|
print("\n" + "=" * 50)
|
|
|
print("📊 Test Results Summary:")
|
|
|
print("=" * 50)
|
|
|
|
|
|
all_passed = True
|
|
|
for test_name, result in results:
|
|
|
status = "✅ PASS" if result else "❌ FAIL"
|
|
|
print(f"{status} {test_name}")
|
|
|
if not result:
|
|
|
all_passed = False
|
|
|
|
|
|
if all_passed:
|
|
|
print("\n🎉 All tests passed! Your app is ready for deployment.")
|
|
|
else:
|
|
|
print("\n⚠️ Some tests failed. Please fix the issues before deploying.")
|
|
|
|
|
|
return all_passed
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
success = main()
|
|
|
sys.exit(0 if success else 1)
|
|
|
|
