Seems interesting! Would this tool work with detection/segmentation models like https://github.com/ultralytics/yolov5? Thanks!