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.