原神|在原神里钓鱼,有人竟然用上了深度强化学习,还把它开源了( 二 )


显卡加速:
python fishing.py image -f yolox/exp/yolox_tiny_fish.py -c weights/best_tiny3.pth --conf 0.25 --nms 0.45 --tsize 640 --device gpu
cpu 运行:
python fishing.py image -f yolox/exp/yolox_tiny_fish.py -c weights/best_tiny3.pth --conf 0.25 --nms 0.45 --tsize 640 --device cpu
运行后出现 init ok 后按 r 键开始钓鱼 , 原神需要全屏 。 出于性能考虑检测框不会实时显示 , 处理运算后台进行 。
更多实现细节 , 读者可参考原项目 。