基于yolov5的ai斗地主
参考项目:
https://github.com/tianqiraf/DouZero_For_HappyDouDiZhu
https://github.com/ultralytics/yolov5
环境:
python3.7
torch
当前已有功能: 1. 基于yolov5训练出来的模型进行斗地主牌型识别,如下:
预计开发功能:
- 基于yolov5识别出来的牌的记牌器。
- 替换之前的以cv模板匹配方式进行识别牌型。
新增android 中ncnn调用模型进行推理