File size: 1,297 Bytes
c1f1d32 |
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 |
# Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# Licensed under the NVIDIA Source Code License [see LICENSE for details].
from setuptools import setup, find_packages
# read the contents of your README file
from os import path
this_directory = path.abspath(path.dirname(__file__))
with open(path.join(this_directory, 'README.md'), encoding='utf-8') as f:
lines = f.readlines()
# remove images from README
lines = [x for x in lines if (('.png' not in x) and ('.gif' not in x))]
long_description = ''.join(lines)
setup(
name="mimicgen",
packages=[
package for package in find_packages() if package.startswith("mimicgen")
],
install_requires=[
"numpy>=1.13.3",
"h5py",
"tqdm",
"imageio",
"imageio-ffmpeg",
"gdown",
"chardet",
"huggingface_hub",
],
eager_resources=['*'],
include_package_data=True,
python_requires='>=3',
description="MimicGen: A Data Generation System for Scalable Robot Learning using Human Demonstrations",
author="Ajay Mandlekar",
url="https://github.com/NVlabs/mimicgen",
author_email="[email protected]",
version="1.0.0",
long_description=long_description,
long_description_content_type='text/markdown'
)
|