如何在 Google Colab 上运行 Samurai

!pip install matplotlib==3.7 tikzplotlib jpeg4py opencv-python lmdb pandas scipy loguru

!git clone https://github.com/yangchris11/samurai.git

%cd samurai/sam2
!pip install -e .
!pip install -e ".[notebooks]"

%cd /content/samurai/sam2/checkpoints
!./download_ckpts.sh && \
%cd ..

import os

# define the data directory
data_directory = '/content/samurai/data/lasot'

# create the data directory if it does not exist
try:
    os.makedirs(data_directory, exist_ok=true)
    print(f"directory '{data_directory}' created successfully or already exists.")
except oserror as error:
    print(f"error creating directory '{data_directory}': {error}")

# define the content to be written to the file
content = '''cat-1
cat-20'''

# define the file path
file_path = os.path.join(data_directory, 'testing_set.txt')

# write the content to the file
try:
    with open(file_path, 'w') as f:
        f.write(content)
    print(f"content written to file '{file_path}' successfully.")
except ioerror as error:
    print(f"error writing to file '{file_path}': {error}")

# print the file path
print(f'file path: {file_path}')

import os
from huggingface_hub import hf_hub_download
import zipfile
import shutil

def download_and_extract(base_dir="/content/samurai/data"):
    try:
        # create lasot and cat directories
        lasot_dir = os.path.join(base_dir, "lasot")
        cat_dir = os.path.join(lasot_dir, "cat")
        os.makedirs(cat_dir, exist_ok=true)

        # create directory to save the zip file
        zip_dir = os.path.join(base_dir, "zips")
        os.makedirs(zip_dir, exist_ok=true)

        print("downloading dataset...")
        zip_path = hf_hub_download(
            repo_id="l-lt/lasot",
            filename="cat.zip",
            repo_type="dataset",
            local_dir=zip_dir
        )
        print(f"downloaded to: {zip_path}")

        # extract zip file to cat directory
        print("extracting zip file to cat directory...")
        with zipfile.zipfile(zip_path, 'r') as zip_ref:
            zip_ref.extractall(cat_dir)

        print("\ncreated directory structure:")
        print("lasot/")
        print("└── cat/")
        # display the first few cat folders
        for item in sorted(os.listdir(cat_dir))[:6]:
            print(f"    ├── {item}/")
        print("    └── ...")

        return lasot_dir

    except exception as e:
        print(f"an error occurred: {str(e)}")
        return none

if __name__ == "__main__":
    extract_path = download_and_extract()
    if extract_path:
        print("\ndownload and extraction completed successfully!")
    else:
        print("\ndownload and extraction failed.")

%cd /content/samurai
!python scripts/main_inference.py