CoolFace
Modelpublic

cpnlab/YOLOR-Streetlights

sourceHugging Faceupdated 4mo agoView on Hugging Face
0likes132downloads
Model Card

YOLOR-Streetlights

PyTorch YOLOv11 Streetlight arXiv Venue

<table> <tr> <td width="30%" valign="top"> <img src="streetlight.png" alt="YOLOR-Streetlights — example streetlight detection" width="100%"> </td> <td valign="top">

YOLOR-Streetlights is a fine-tuned object detection model for BS identification for beam initialization to detect urban streetlight infrastructure in one inference pass. Data was collected on the [University of Nebraska–Lincoln](https://www.unl.edu/) campus.

</td> </tr> </table>

Reference implementation for the paper:

Avhishek Biswas\, Apala Pramanik\, Eylem Ekici, Mehmet C. Vuran. "Look Once, Beam Twice: Camera-Primed Real-Time Double-Directional mmWave Beam Management for Vehicular Connectivity." (\*equal contribution) arXiv: <https://doi.org/10.48550/arXiv.2605.05071>

<p align="center"> <img src="overview2_updated.png" alt="VIBE five-stage camera-primed beam-management pipeline" width="90%"> </p>

Quick links

  • Paper (arXiv): <https://doi.org/10.48550/arXiv.2605.05071>
  • Code and Data: <https://github.com/UNL-CPN-Lab/Look-Once-Beam-Twice>
  • Training pipeline: <https://github.com/UNL-CPN-Lab/Look-Once-Beam-Twice/tree/main/YOLOR_Training>
ArchitectureYOLOv11x, 81-class output head (COCO 80 + 1 custom)
Initializationstock yolo11x.pt
Schedule200 epochs, cos_lr, close_mosaic=20, lr0=0.01
Training dataStreetlights — 1,498 train / 166 valid / 182 test
Custom classesstreetlight (id 80)
Released checkpointlast.pt

Usage

python
from huggingface_hub import hf_hub_download
from ultralytics import YOLO

weights = hf_hub_download(repo_id="cpnlab/YOLOR-Streetlights", filename="last.pt")
model = YOLO(weights)
results = model.predict("path/to/image.jpg", conf=0.25)

Class indices: 0–79 = COCO; 80 = streetlight.

Training data

Code and Data: <https://github.com/UNL-CPN-Lab/Look-Once-Beam-Twice>

Citation

bibtex
@inproceedings{biswas2026look,
  title     = {Look Once, Beam Twice: Camera-Primed Real-Time Double-Directional
               mmWave Beam Management for Vehicular Connectivity},
  author    = {Biswas, Avhishek and Pramanik, Apala and Ekici, Eylem and Vuran, Mehmet C.},
  booktitle = {Proc. IEEE SECON},
  year      = {2026}
}

Paper: <https://doi.org/10.48550/arXiv.2605.05071>

Contact

For questions about this model or the paper, contact the corresponding authors:

Acknowledgments

Developed at the [Cyber Physical Networking (CPN) Lab](https://cpn.unl.edu/), School of Computing, University of Nebraska–Lincoln, in collaboration with The Ohio State University. Thanks to Sivers Semiconductors, Ettus Research, and the open-source Ultralytics, PyTorch, and Ettus UHD communities.