FilesExpand file tree

 master

/

setup.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

61 lines (52 loc) · 2.03 KB

 master

/

setup.py

Copy path

Top

File metadata and controls

• Code
• Blame

61 lines (52 loc) · 2.03 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

61

from setuptools import setup, find_packages

import sys

import platform

python_version = platform.python_version().rsplit('.', maxsplit=1)[0]

mac_v, _, _ = platform.mac_ver()

if mac_v != '':

mac_version = '.'.join(mac_v.split('.')[:2])

else:

mac_version = None

requirements = [

"pillow",

"requests",

"numpy==1.19.3",

"tensorflow==2.4;platform_machine!='armv7l'",

"onnxruntime==1.6.0;platform_machine!='armv7l'"

]

# get the right TF Lite runtime packages based on OS and python version: https://www.tensorflow.org/lite/guide/python#install_just_the_tensorflow_lite_interpreter

tflite_python = None

tflite_machine = None

# get the right python string for the version

if python_version == '3.5':

tflite_python = 'cp35-cp35m'

elif python_version == '3.6':

tflite_python = 'cp36-cp36m'

elif python_version == '3.7':

tflite_python = 'cp37-cp37m'

elif python_version == '3.8':

tflite_python = 'cp38-cp38'

# get the right machine string

if sys.platform == 'win32':

tflite_machine = 'win_amd64'

elif sys.platform == 'darwin' and mac_version == '10.15':

tflite_machine = 'macosx_10_15_x86_64'

elif sys.platform == 'linux':

if platform.machine() == 'x86_64':

tflite_machine = 'linux_x86_64'

elif platform.machine() == 'armv7l':

tflite_machine = 'linux_armv7l'

# add it to the requirements, or print the location to find the version to install

if tflite_python and tflite_machine:

requirements.append(f"tflite_runtime @ https://github.com/google-coral/pycoral/releases/download/release-frogfish/tflite_runtime-2.5.0-{tflite_python}-{tflite_machine}.whl")

else:

print(

f"Couldn't find tflite_runtime for your platform {sys.platform}, machine {platform.machine()}, and python version {python_version}, please see the install guide for the right version: https://www.tensorflow.org/lite/guide/python#install_just_the_tensorflow_lite_interpreter"

)

setup(

name="lobe",

version="0.3.0",

packages=find_packages("src"),

package_dir={"": "src"},

install_requires=requirements,

)