Note: 因為上傳代碼時匆忙,我沒有認真審核各部分的代碼,不確定能不能跑(手邊沒Windows系統的電腦,而項目內用到了Windows的接口)且整個項目代碼很冗余。計划21年底重寫一遍代碼(計划用pytorch,及更換數據集和網絡,如果沒鴿的話)
說明
- 該項目參考(其實是直接拿來用)https://github.com/Sentdex/pygta5
B站視頻鏈接
https://www.bilibili.com/video/BV1Th41167u7
教程
- 配置環境
- 這里推薦一下通過conda直接裝GPU版的tensorflow,輸入命令:
conda install tensorflow-gpu=2.3 tensorflow=2.3=mkl_py37h936c3e2_0
- 安裝
keras,tfleran,mss,pywin32,opencv
等等庫,大家運行后按代碼提示安裝包即可
- 這里推薦一下通過conda直接裝GPU版的tensorflow,輸入命令:
- 下載數據
- 百度網盤鏈接:鏈接: https://pan.baidu.com/s/1RB7rl6EkjNATzxEbEG25Cg 密碼: al32
train_model.py
是訓練模型的,如果想基於已有的模型再訓練,可以修改字段LOAD_MODEL = True
為LOAD_MODEL = False
- 如果想自己訓練,那么把
pygta5-motorcycle-training-data-and-model.zip
解壓后,把training_data
中的npy文件放到datasets
目錄下
- 如果想自己訓練,那么把
GRAD_SCREEN.py
是用來測試模型的- 把GTA5游戲分辨率設置為
800*600
,並把游戲窗口放到左上角 - 直接運行
GRAD_SCREEN.py
- 回到游戲窗口,點擊一下即可。 如果想暫停程序,按大寫
T
,然后ctrl+c強制取消運行,不然會瘋狂輸入東西(doge) - 其實
pygta5-motorcycle-training-data-and-model.zip
包含原作者訓練好的模型,但是在我這邊使用模型很垃圾,不知道為什么,如果大家用我訓練的模型也很垃圾可以試試原作者的模型,方法是:修改model.load(r"./model_saved/alexnet_num_3-gta-v-1-epochs-1000-lr-0.001.tflearn",weights_only=True)
變成model.load(r"解壓后zip/model/pygta5-car-fast-0.001-alexnetv2-10-epochs-300K-data.model.data-00000-of-00001", weights_only=True)
- 把GTA5游戲分辨率設置為