86935e81ea
* fix version of pifpaf and nuscenes * set version of all dependancies * add iccv version * fix readme * fix readme and tabulate version * fix pylint and pytest version * update to pifpaf 0.9 * fix pylint and pytorch versions * fix pifpaf=0.8 * fix pytorch version * make new version * fix bug on commas * change to torch 1.1 * change pylint version
45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
from setuptools import setup
|
|
|
|
# extract version from __init__.py
|
|
with open('monoloco/__init__.py', 'r') as f:
|
|
VERSION_LINE = [l for l in f if l.startswith('__version__')][0]
|
|
VERSION = VERSION_LINE.split('=')[1].strip()[1:-1]
|
|
|
|
setup(
|
|
name='monoloco',
|
|
version=VERSION,
|
|
packages=[
|
|
'monoloco',
|
|
'monoloco.network',
|
|
'monoloco.eval',
|
|
'monoloco.train',
|
|
'monoloco.prep',
|
|
'monoloco.visuals',
|
|
'monoloco.utils'
|
|
],
|
|
license='GNU AGPLv3',
|
|
description='MonoLoco: Monocular 3D Pedestrian Localization and Uncertainty Estimation',
|
|
long_description=open('README.md').read(),
|
|
long_description_content_type='text/markdown',
|
|
author='Lorenzo Bertoni',
|
|
author_email='lorenzo.bertoni@epfl.ch',
|
|
url='https://github.com/vita-epfl/monoloco',
|
|
zip_safe=False,
|
|
|
|
install_requires=[
|
|
'torch==1.1.0',
|
|
'torchvision==0.3.0',
|
|
'openpifpaf==0.8.0',
|
|
'tabulate==0.8.3', # For evaluation
|
|
],
|
|
extras_require={
|
|
'test': [
|
|
'pylint==2.4.2',
|
|
'pytest==4.6.3',
|
|
],
|
|
'prep': [
|
|
'nuscenes-devkit==1.0.2',
|
|
],
|
|
},
|
|
)
|