I'm not a professional at A.I but i highly recommend using yolov5, It's available on github and there's plenty of documentation and YT videos on It You can clone it, install the requirments and have fun, you can even use live camera!