forked from NVlabs/ocropus3-ocrorot
-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
38 lines (32 loc) · 994 Bytes
/
setup.py
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
#!/bin/env python
# Copyright (c) 2017 NVIDIA CORPORATION. All rights reserved.
# See the LICENSE file for licensing terms (BSD-style).
from __future__ import print_function
import os
import sys
import urllib
import os.path
from distutils.core import setup # , Extension, Command
assert sys.version_info[0] == 2 and sys.version_info[1] >= 7,\
"requires Python version 2.7 or later, but not Python 3.x"
remote = "https://storage.googleapis.com/tmb-models/"
models = ["rot-000003456-020897.pt", "logskew-000015808-000132.pt"]
for model in models:
if not os.path.exists(model):
print("downloading", remote+model)
urllib.URLopener().retrieve(remote+model, model)
scripts = """
ocrorot-train
ocroskew-train
ocrorot-pred
ocroskew-pred
""".split()
setup(
name='ocrorot',
version='v0.0',
author="Thomas Breuel",
description="Page rotation detection/correction.",
packages=["ocrorot"],
scripts=scripts,
data_files=[("share/ocrorot", models)]
)