machine-learning-flying-birds/PATH.py at master · liuzimo/machine-learning-flying-birds · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#coding:utf-8
import cv2
import numpy as np
def load():
    # 小鸟图片的载入路径
    PlayerPath = (
            './assets/sprites/redbird-upflap.png',
            './assets/sprites/redbird-midflap.png',
            './assets/sprites/redbird-downflap.png'
    )

    # 背景图片
    BackgroundPath = './assets/sprites/background-black.png'

    # 管道障碍物图片
    PipePath = './assets/sprites/pipe-green.png'

    Images = {}

    # 数字
    Images['numbers'] = (
        cv2.imread('./assets/sprites/0.png',-1),
        cv2.imread('./assets/sprites/1.png',-1),
        cv2.imread('./assets/sprites/2.png',-1),
        cv2.imread('./assets/sprites/3.png',-1),
        cv2.imread('./assets/sprites/4.png',-1),
        cv2.imread('./assets/sprites/5.png',-1),
        cv2.imread('./assets/sprites/6.png',-1),
        cv2.imread('./assets/sprites/7.png',-1),
        cv2.imread('./assets/sprites/8.png',-1),
        cv2.imread('./assets/sprites/9.png',-1)
    )

    # 地面
    Images['base'] = cv2.imread('./assets/sprites/base.jpg')

    Sounds = {}


    Images['background'] = np.zeros((512,288,3),dtype=np.uint8)
    Images['background'][400:512,0:288] = Images['base']
    #随机选取小鸟的样子
    Images['player'] = (
        cv2.imread(PlayerPath[0],-1),
        cv2.imread(PlayerPath[1],-1),
        cv2.imread(PlayerPath[2],-1),
    )

    # 选取管道障碍物
    Images['pipe'] = (
        cv2.flip(cv2.imread(PipePath),-1),
        cv2.imread(PipePath),
    )


    return Images


if __name__ == '__main__':
    load()