A tool to scan Android devices for ML model weight files via ADB.
Install
pip install thebraincollector
Usage
from tbc import find_ml_models, get_device_info # Get device information info = get_device_info() print(f"Device: {info['model']}") # Find ML models on connected device models = find_ml_models()
CLI
tbc --help tbc --file /path/to/file tbc --local-dir /path/to/dir tbc --export-csv results.csv --cleanup
API
get_device_info() -> dict
Returns device serial number, model, and manufacturer.
find_ml_models() -> set
Searches connected Android device for ML model files.
scan_files(files: list[str], local: bool = False) -> None
Scans files for ML model or APK extensions.
human_readable_size(size_bytes: int) -> str
Converts bytes to human-readable format.
calculate_md5(file_path: str) -> str | None
Calculates MD5 hash of a file.
Development
git clone https://github.com/dclavijo/TBC.git cd TBC pip install -e ".[test]" # run tests pytest # format ruff format src/ tests/ # lint ruff check src/ tests/ # type check mypy src/
License
MIT License - see LICENSE file for details.