Skip to content

Latest commit

 

History

History
15 lines (10 loc) · 798 Bytes

README.md

File metadata and controls

15 lines (10 loc) · 798 Bytes
Raw

Kalman YOLO

Model weights and configs for yolo_ros.

Repository Structure

  • config: Configuration files for different competitions.
  • models: Pre-trained YOLO models in PyTorch .pt format.
  • launch: A launch file for this package. Accepts arguments:
    • rgbd_ids=(space separated camera names, e.g. d455_front)
    • config=(name of config file without .yaml suffix)

Training a Model

Use the yolo-gym repository to train a model and export the PyTorch weight here to the models directory. Then create a new config file in the config directory. Remember to update the parameters, in particular: model, class_names, class_radii.