FilesExpand file tree

 master

/

download_prebuilt_dependencies.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

62 lines (52 loc) · 2.45 KB

 master

/

download_prebuilt_dependencies.py

Copy path

Top

File metadata and controls

• Code
• Blame

62 lines (52 loc) · 2.45 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

#!/usr/bin/python

#

# copyright Guillaume Dumont (2016)

import os

import sys

import hashlib

import argparse

import tarfile

from six.moves import urllib

from download_model_binary import reporthook

WIN_DEPENDENCIES_URLS = {

('v120', '2.7'):("https://github.com/Coderx7/Caffe_1.0_Windows/releases/download/caffe_1.0_windows/libraries_v120_x64_py27_1.1.0.tar.bz2",

"ba833d86d19b162a04d68b09b06df5e0dad947d4"),

('v140', '2.7'):("https://github.com/Coderx7/Caffe_1.0_Windows/releases/download/caffe_1.0_windows/libraries_v140_x64_py27_1.1.0.tar.bz2",

"17eecb095bd3b0774a87a38624a77ce35e497cd2"),

('v140', '3.5'):("https://github.com/Coderx7/Caffe_1.0_Windows/releases/download/caffe_1.0_windows/libraries_v140_x64_py35_1.1.0.tar.bz2",

"f060403fd1a7448d866d27c0e5b7dced39c0a607"),

('v140', '3.6'):("https://github.com/Coderx7/Caffe_1.0_Windows/releases/download/caffe_1.0_windows/libraries_v140_x64_py36_1.1.0.tar.bz2",

"2676374a5179a8101ba5f1c5da63bba31df109a8"),

}

# function for checking SHA1.

def model_checks_out(filename, sha1):

with open(filename, 'rb') as f:

return hashlib.sha1(f.read()).hexdigest() == sha1

if __name__ == '__main__':

parser = argparse.ArgumentParser(

description='Download prebuilt dependencies for windows.')

parser.add_argument('--msvc_version', default='v120', choices=['v120', 'v140'])

args = parser.parse_args()

# get the appropriate url

pyver = '{:d}.{:d}'.format(sys.version_info.major, sys.version_info.minor)

try:

url, sha1 = WIN_DEPENDENCIES_URLS[(args.msvc_version, pyver)]

except KeyError:

print('ERROR: Could not find url for MSVC version = {} and Python version = {}.\n{}'

.format(args.msvc_version, pyver,

'Available combinations are: {}'.format(list(WIN_DEPENDENCIES_URLS.keys()))))

sys.exit(1)

dep_filename = os.path.split(url)[1]

# Download binaries

print("Downloading dependencies ({}). Please wait...".format(dep_filename))

urllib.request.urlretrieve(url, dep_filename, reporthook)

if not model_checks_out(dep_filename, sha1):

print('ERROR: dependencies did not download correctly! Run this again.')

sys.exit(1)

print("\nDone.")

# Extract the binaries from the tar file

tar = tarfile.open(dep_filename, 'r:bz2')

print("Extracting dependencies. Please wait...")

tar.extractall()

print("Done.")

tar.close()