File size: 1,822 Bytes
d670799 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
#!/usr/bin/env python3
"""
Script to download the TSN model checkpoint for GenVidBench
"""
import os
import requests
from tqdm import tqdm
def download_file(url, filename):
"""Download a file with progress bar"""
response = requests.get(url, stream=True)
total_size = int(response.headers.get('content-length', 0))
with open(filename, 'wb') as f, tqdm(
desc=filename,
total=total_size,
unit='iB',
unit_scale=True,
unit_divisor=1024,
) as pbar:
for chunk in response.iter_content(chunk_size=8192):
size = f.write(chunk)
pbar.update(size)
def main():
"""Download the TSN model checkpoint"""
# Create checkpoints directory
os.makedirs('checkpoints', exist_ok=True)
# Model checkpoint URL (you may need to update this with the actual URL)
checkpoint_url = "https://download.openmmlab.com/mmaction/v1.0/recognition/tsn/tsn_r50_8xb32-1x1x8-100e_kinetics400-rgb/tsn_r50_8xb32-1x1x8-100e_kinetics400-rgb_20220818-2692d16c.pth"
checkpoint_path = "checkpoints/tsn_r50_8xb32-1x1x8-100e_kinetics400-rgb_20220818-2692d16c.pth"
if os.path.exists(checkpoint_path):
print(f"Checkpoint already exists: {checkpoint_path}")
return
print(f"Downloading TSN model checkpoint...")
print(f"URL: {checkpoint_url}")
print(f"Destination: {checkpoint_path}")
try:
download_file(checkpoint_url, checkpoint_path)
print(f"✅ Successfully downloaded checkpoint to {checkpoint_path}")
except Exception as e:
print(f"❌ Error downloading checkpoint: {e}")
print("Please download the checkpoint manually and place it in the checkpoints/ directory")
if __name__ == "__main__":
main()
|