Add NVIDIA driver update scripts and GPU verification

.pre-commit-config.yaml

@@ -37,21 +37,37 @@ repos:
         stages: [pre-commit]
         pass_filenames: false
 
-  - repo: https://github.com/kynan/nbstripout
-    rev: 0.8.1
-    hooks:
-      - id: nbstripout
-
-  # Zuban type checker
+  # Zuban type checker (scoped to a single directory)
   - repo: local
     hooks:
       - id: zuban
-        name: Zuban type checker
-        entry: uv run zuban check
+        name: Zuban type checker (visdet/structures)
+        entry: uv run zuban check visdet/structures
+        language: system
+        types: [python]
+        pass_filenames: false
+        files: ^visdet/structures/
+
+      - id: zuban-apis
+        name: Zuban type checker (visdet/apis)
+        entry: uv run zuban check visdet/apis
         language: system
         types: [python]
         pass_filenames: false
-        exclude: ^(archive|tests|configs)/
+        files: ^visdet/apis/
+
+      - id: zuban-core-mask
+        name: Zuban type checker (visdet/core/mask)
+        entry: uv run zuban check visdet/core/mask
+        language: system
+        types: [python]
+        pass_filenames: false
+        files: ^visdet/core/mask/
+
+  - repo: https://github.com/kynan/nbstripout
+    rev: 0.8.1
+    hooks:
+      - id: nbstripout
 
   # Block old-style viscv/visengine imports
   - repo: local

README.md

@@ -4,6 +4,7 @@
 
   <p>
     <a href="https://pypi.org/project/visdet"><img src="https://img.shields.io/pypi/v/visdet?style=flat-square&color=blue" alt="PyPI"/></a>
+    <a href="https://github.com/BinItAI/visdet/actions/workflows/test.yml"><img src="https://img.shields.io/github/actions/workflow/status/BinItAI/visdet/test.yml?branch=main&style=flat-square" alt="Tests"/></a>
     <a href="https://binitai.github.io/visdet/"><img src="https://img.shields.io/badge/docs-latest-brightgreen?style=flat-square" alt="Documentation"/></a>
   </p>
 

configs/presets/datasets/coco_instance_segmentation.yaml

@@ -9,12 +9,12 @@ type: CocoDataset
 data_root: data/coco/
 ann_file: annotations/instances_train2017.json
 data_prefix:
-  img: train2017/
+  img_path: train2017/
 
 # Validation paths
 val_ann_file: annotations/instances_val2017.json
 val_data_prefix:
-  img: val2017/
+  img_path: val2017/
 
 # Data pipeline for training
 train_pipeline:

docs/getting-started/quick-start.md

@@ -2,19 +2,36 @@
 
 This guide provides quick examples to get you started with VisDet.
 
+## Training with SimpleRunner
+
+The easiest way to train a model is using the `SimpleRunner` API:
+
+```python
+from visdet import SimpleRunner
+
+# Simple, string-based API - just like Hugging Face or Ultralytics
+runner = SimpleRunner(
+    model='mask_rcnn_swin_s',
+    dataset='coco_instance_segmentation',
+    epochs=12,
+    batch_size=2
+)
+
+runner.train()
+```
+
 ## Inference with Pre-trained Models
 
 ### Using High-level APIs
 
 You can use high-level APIs to perform inference on images:
 
 ```python
-from mmdet.apis import init_detector, inference_detector
-import mmcv
+from visdet.apis import init_detector, inference_detector
 
 # Specify the config file and checkpoint file
 config_file = 'configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py'
-checkpoint_file = 'checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth'
+checkpoint_file = 'checkpoints/faster_rcnn_r50_fpn_1x_coco.pth'
 
 # Build the model from a config file and a checkpoint file
 model = init_detector(config_file, checkpoint_file, device='cuda:0')
@@ -24,7 +41,7 @@ img = 'demo/demo.jpg'
 result = inference_detector(model, img)
 
 # Show the results
-from mmdet.apis import show_result_pyplot
+from visdet.apis import show_result_pyplot
 show_result_pyplot(model, img, result)
 ```
 

docs/index.md

@@ -33,6 +33,13 @@ The master branch works with **PyTorch 1.5+**.
 
 </details>
 
+## Roadmap
+
+See our [Roadmap](roadmap.md) for planned features, including:
+- **SPDL Integration**: Thread-based data loading for 74% faster training
+- **Kornia**: GPU-accelerated augmentations
+- **Python 3.13t**: Free-threaded Python support
+
 ## What's New
 
 For the latest updates and improvements to VisDet, please refer to the [changelog](about/changelog.md).