rajeshpt

@Jacobsolawetz thanks for your quick response. Just as a suggestion, would it make sense not to keep a copy of the yolo source code and instead just use the torch.hub.load...

> @rajeshpt interesting... can you use torch.hub.load with models that you have trained on your custom dataset? yes, you can. I am currently doing that and that is the recommended...

Loading your custom model - model = torch.hub.load("ultralytics/yolov5", "custom", MODEL_PATH) for Inference - results = model(frame) You can look at this for more details - https://github.com/ultralytics/yolov5/issues/36