Can candle-yolo do training and inference?

⚓ Rust    📅 2026-02-05    👤 surdeus    👁️ 6      

surdeus

Hi guys I have some questions, so if I understand correctly candle is a replacement of pytorch and can be used with yolo to train images? I assume you use candle-yolo to do training for a custom model?

I also wanted to ask can candle run models after training on images, videos and live cameras, similar to this code for yolo/ultralytics with python:

from ultralytics import YOLO

model = YOLO("custom_ncnn_model/")

model.predict(source = "video.mp4",
show = True, conf = 0.6, line_thickness = 2, save = False)

1 post - 1 participant

Read full topic

🏷️ Rust_feed