|
|
|
|
|
|
|
|
|
|
|
import re
|
|
|
import setuptools
|
|
|
import glob
|
|
|
from os import path
|
|
|
import torch
|
|
|
from torch.utils.cpp_extension import CppExtension
|
|
|
|
|
|
torch_ver = [int(x) for x in torch.__version__.split(".")[:2]]
|
|
|
assert torch_ver >= [1, 3], "Requires PyTorch >= 1.3"
|
|
|
|
|
|
|
|
|
def get_extensions():
|
|
|
this_dir = path.dirname(path.abspath(__file__))
|
|
|
extensions_dir = path.join(this_dir, "yolox", "layers", "csrc")
|
|
|
|
|
|
main_source = path.join(extensions_dir, "vision.cpp")
|
|
|
sources = glob.glob(path.join(extensions_dir, "**", "*.cpp"))
|
|
|
|
|
|
sources = [main_source] + sources
|
|
|
extension = CppExtension
|
|
|
|
|
|
extra_compile_args = {"cxx": ["-O3"]}
|
|
|
define_macros = []
|
|
|
|
|
|
include_dirs = [extensions_dir]
|
|
|
|
|
|
ext_modules = [
|
|
|
extension(
|
|
|
"yolox._C",
|
|
|
sources,
|
|
|
include_dirs=include_dirs,
|
|
|
define_macros=define_macros,
|
|
|
extra_compile_args=extra_compile_args,
|
|
|
)
|
|
|
]
|
|
|
|
|
|
return ext_modules
|
|
|
|
|
|
|
|
|
with open("yolox/__init__.py", "r") as f:
|
|
|
version = re.search(
|
|
|
r'^__version__\s*=\s*[\'"]([^\'"]*)[\'"]',
|
|
|
f.read(), re.MULTILINE
|
|
|
).group(1)
|
|
|
|
|
|
|
|
|
with open("README.md", "r") as f:
|
|
|
long_description = f.read()
|
|
|
|
|
|
|
|
|
setuptools.setup(
|
|
|
name="yolox",
|
|
|
version=version,
|
|
|
author="basedet team",
|
|
|
python_requires=">=3.6",
|
|
|
long_description=long_description,
|
|
|
ext_modules=get_extensions(),
|
|
|
classifiers=["Programming Language :: Python :: 3", "Operating System :: OS Independent"],
|
|
|
cmdclass={"build_ext": torch.utils.cpp_extension.BuildExtension},
|
|
|
packages=setuptools.find_namespace_packages(),
|
|
|
)
|
|
|
|
