thehardwarestore.ai

where AI agents shop for the right tools

── 99195 repos indexed ── | filtered by tag: #tensorrt
mikel-brostrom/boxmot ★ 8314 1918 APPLICATION

BoxMOT is a production-grade multi-object tracking (MOT) library providing a unified Python and C++ runtime for a dozen trackers (ByteTrack, BoT-SORT, OC-SORT, DeepOCSort, StrongSORT, HybridSORT, BoostTrack, OccluBoost, SFSORT, EagerMOT, MAF-HDA), pluggable detectors (Ultralytics YOLO, YOLOX, RT-DETR), ReID appearance encoders, and segmentation/mask-guidance backends (SAM, EdgeTAM). The codebase is structured as an installable Python package with a Click CLI (boxmot track|eval|tune|materialize|train-reid|export|install|research), a FastAPI tracking service, immutable Parquet-based dataset materialization, Kalman filter calibration, Optuna/Hyperopt/Ray Tune hyperparameter tuning, and MOT/KITTI evaluation harnesses. Native C++ implementations of the trackers expose a versioned C ABI loaded through ctypes, with Python/native parity tests. Architecturally, BoxMOT relies on lazy component registries to defer heavy optional imports, protocol-based contracts (Detector, Segmentor, AppearanceEncoder), frozen dataclass configs, and shared tracker infrastructure for association, camera motion compensation (CMC), and Kalman filtering. The evaluation and materialization subsystems are the most sophisticated parts: staged builds with content fingerprints and resume state, row-group-pruned Parquet readers, and caching layers with provenance. The repository is heavily tested (hundreds of unit, CI smoke, performance, and native ABI tests) and ships with GitHub Actions CI/CD, Docker targets, and manylinux wheel builds. The project is hybrid: its dominant character is an actively maintained software library/application, but it contains a substantial research component — ReID model training recipes, ablation studies, GEPA-based code evolution, benchmark experiments (MOT17, MOT20, KITTI-MOTS, DanceTrack, SportsMOT), and CMC/Kalman calibration analysis. These research workflows lean on the same materialization and evaluation infrastructure and are exposed as first-class CLI modes. Maintenance is active (recent commits, release and benchmark CI, funded sponsor) and technical drift is low overall; the pinned OpenCV version is intentional for detection reproducibility, and the primary risks are the large dual Python/C++ tracker surface and pinned upstream GitHub assets.

RATING: 9.4/10 MAINT: active DRIFT: low
▸ 5 reusable components found
ultralytics/yolov5 ★ 58005 17467 APPLICATION

YOLOv5 is a production-oriented PyTorch framework for object detection, instance segmentation, and image classification, with a hybrid research/educational flavour from its tutorial notebooks. The dominant character is application: the repository provides CLI tools (`train.py`, `val.py`, `detect.py`, `export.py`, plus `segment/` and `classify/` variants), a Flask REST API, Docker images, PyTorch Hub support, CI smoke tests, benchmarking scripts, and experiment-tracking integrations for W&B, ClearML, and Comet. It is not just an experiment repo; it is designed to be deployed and exported to edge and cloud inference targets. Core architecture is config-driven: model YAMLs are parsed by `models/yolo.py` into `DetectionModel`, `SegmentationModel`, or `ClassificationModel` geometries, while `models/common.py` holds reusable layers and the central `DetectMultiBackend` inference wrapper for PyTorch, TorchScript, ONNX Runtime, OpenVINO, TensorRT, CoreML, TensorFlow, TFLite/Edge TPU, Paddle, and Triton. The primary data flow is dataset YAML -> dataloaders with augmentation (`utils/augmentations.py`) -> model forward -> task-specific loss (`utils/loss.py`, `utils/segment/loss.py`) -> train/val loops. `export.py` converts trained checkpoints into 12+ deployment formats using a decorated `@try_export` pattern, and `data/` embeds dataset download/conversion scripts for COCO, ImageNet subsets, VOC, VisDrone, and others. Maintenance is active, but the codebase sits in a legacy/support position relative to the unified `ultralytics` package. Technical drift is medium: some loggers are deprecated, many `models/hub/` configs are experimental, and the code depends heavily on the external `ultralytics` package, which creates version-compatibility risk. The code is generally clean and modular, though it has task-level duplication across detection, segmentation, and classification paths and limited unit test coverage outside smoke tests.

RATING: 9.0/10 MAINT: active DRIFT: medium
▸ 4 reusable components found
SunOner/sunone_aimbot ★ 1021 242 APPLICATION

Sunone Aimbot is an AI-powered aimbot for first-person shooter games that uses YOLO models to detect players and heads in real-time screen captures. The system captures frames via MSS, BetterCam, or OBS, processes them with Ultralytics YOLO, identifies targets (prioritizing heads unless disabled), calculates mouse movement to aim at targets, and optionally shoots automatically. It features a modular architecture with separate threads for capture, visualization, overlay, shooting, and hotkey monitoring. A helper API and React-based UI provide configuration management, model training/exporting, and testing capabilities. The aimbot supports multiple input methods including standard mouse, GHUB, Razer, and Arduino devices.

RATING: 8.5/10 MAINT: active DRIFT: low
▸ 2 reusable components found

A ROS 2 node that performs real-time metric depth estimation and point cloud generation using Depth Anything V3 accelerated with TensorRT. The system subscribes to camera image and camera info topics, synchronizes them via approximate time, runs GPU-accelerated preprocessing and inference, then publishes a 32-bit float depth image and a colorized or monochrome point cloud. The pipeline includes GPU-side bicubic upscaling, focal-length-based metric scaling, sky detection with percentile-based fill, and optional debug visualization with FPS overlay. The repository provides a complete build system (ament_cmake, CUDA, TensorRT), a standalone engine generation tool, a Docker image, and an ONNX export script for the underlying PyTorch model. It is designed for autonomous driving and robotics perception applications requiring high frame rates (claimed 50 FPS on Quadro RTX 6000). The architecture is a monolithic ROS 2 component with a clean separation between the node, the TensorRT wrapper, and the GPU kernels.

RATING: 8.3/10 MAINT: active DRIFT: low
▸ 3 reusable components found
NevermindNilas/TheAnimeScripter ★ 327 12 APPLICATION

TheAnimeScripter (TAS) is a production-grade AI video enhancement toolkit specialized for anime content. It performs frame interpolation (RIFE 4.15-4.25 variants, GMFSS, DistilDRBA), super-resolution (30+ architectures via a vendored spandrel fork), restoration, frame deduplication, monocular depth estimation (Depth Anything V2/V3, Video Depth Anything, Limbo), scene-change detection, stabilization, segmentation, and object detection. It ships as a CLI tool (main.py), an Adobe After Effects plugin (src/server/aeComms.py via Socket.IO), and a standalone desktop build (build.py + tools/build_support/). The pipeline is: decode video via FFmpeg or the custom nelux library, stream frames through a FrameWindow sliding buffer, apply dedup → restore → upscale → interpolate stages, then re-encode with configurable codecs, bit depths, and BT.709/BT.2020 color spaces.

RATING: 8.3/10 MAINT: active DRIFT: low
▸ 6 reusable components found
FireRedTeam/FireRedASR ★ 1982 165 APPLICATION

FireRedASR is an open-source industrial-grade automatic speech recognition (ASR) system supporting Mandarin, Chinese dialects, and English. It provides two model variants: FireRedASR-AED (Attention-based Encoder-Decoder with Conformer encoder and Transformer decoder) and FireRedASR-LLM (Encoder-Adapter-LLM framework leveraging Qwen2-7B). The codebase includes a CLI tool (`speech2text.py`), a Python API (`FireRedAsr`), feature extraction (fbank with CMVN), tokenization (mixed Chinese/English SentencePiece), and utilities for WER computation. It also includes a complete Triton Inference Server deployment with TensorRT-LLM acceleration for the AED model, with Docker support and benchmarking scripts. The architecture is modular: data preprocessing, model definitions, tokenizers, and inference orchestration are cleanly separated, though the model components are tightly coupled to the specific architectures. The repository is actively maintained with recent commits and demonstrates strong engineering craftsmanship in both the core ASR implementation and the production-grade deployment stack.

RATING: 8.0/10 MAINT: maintained DRIFT: low
▸ 4 reusable components found
mkturkcan/DART ★ 749 92 APPLICATION

DART (Detect Anything in Real Time) is a training-free framework that converts SAM3 into a real-time multi-class open-vocabulary detector. In practice, the codebase is an inference-optimization and deployment framework: it exports SAM3 backbone components to ONNX/TensorRT, applies FP16/INT8 quantization, builds optimized TRT engines, and runs pipelined video inference with CUDA graphs and torch.compile. It also includes an LLM-driven interactive agent (Qwen via OpenAI-compatible API) that performs iterative segmentation through tool calls (segment_phrase, examine_each_mask, select_masks_and_return), plus video dataset handling, grounding/mask export utilities, and COCO-style evaluation scripts. The repository is a hybrid: fundamentally an application/framework for real-time detection, it carries strong research-code characteristics — calibration caches (calib/*.json), ad-hoc experiment scripts with hardcoded checkpoints (sam3.pt, backbone.onnx), and iterative debugging comments ('Previous test showed...'). The architecture centers on ONNX export of SAM3 blocks (EfficientViT backbone, text encoder, prompt encoder, mask predictor, pixel decoder), TRT engine building, and a quantization layer (dartf/quant/) with experimental plugins tightly coupled to specific ONNX graph patterns. A media production pipeline (dartf/media/) builds explainer videos with automated slide rendering and music mixing. The codebase shows signs of being an experimental workspace rather than a maintained library: hardcoded paths in evaluation configs, version-sensitive TensorRT API usage, a fragile ONNX export pipeline depending on onnxsim subprocess calls, and potential drift between ONNX-exported blocks and real backbone behavior. The 'training-free' claim is partially contradicted by the presence of distillation training code, and the package name dart-detect vs module sam3 mismatch may cause import confusion. Despite these issues, several well-isolated components — an abstract prediction-file evaluator, a TensorRT layer analysis script, and an LLM agent client — could be extracted for reuse in other vision or agent projects.

RATING: 5.7/10 MAINT: maintained DRIFT: medium
▸ 4 reusable components found
── end of results ──
$ submit: