You might want to check out https://github.com/kstenerud/yoloai