|
|
--- |
|
|
license: apache-2.0 |
|
|
pipeline_tag: object-detection |
|
|
--- |
|
|
|
|
|
# Detection Perceiver Model |
|
|
|
|
|
<img src="https://raw.githubusercontent.com/maxploter/DePerceiver/refs/heads/main/de_perceiver_r50/de_perceiver.png" alt="DePerceiver R50 Image"/> |
|
|
|
|
|
[GH repository DePerceiver](https://github.com/maxploter/DePerceiver) |
|
|
|
|
|
## Usage |
|
|
|
|
|
```python |
|
|
from huggingface_hub import snapshot_download |
|
|
import torch |
|
|
import json |
|
|
import sys |
|
|
|
|
|
# Download model |
|
|
repo_path = snapshot_download(repo_id="Max-Ploter/DePerceiver") |
|
|
|
|
|
# Add to Python path to access modeling_detection_perceiver.py |
|
|
sys.path.append(repo_path) |
|
|
from modeling_detection_perceiver import build_model |
|
|
|
|
|
# Load config |
|
|
with open(f"{repo_path}/config.json") as f: |
|
|
config = json.load(f) |
|
|
|
|
|
# Build and load model |
|
|
model = build_model(config) |
|
|
model.load_state_dict(torch.load(f"{repo_path}/detection_perceiver_model_weights_pytorch.bin")) |
|
|
``` |
|
|
|
|
|
## Dependencies |
|
|
|
|
|
```shell |
|
|
pip install torch torchvision perceiver-pytorch |
|
|
``` |
|
|
|