flyhostel/setup.py at master · shaliulab/flyhostel · GitHub

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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
import pathlib
from setuptools import setup, find_packages

import warnings

# The directory containing this file
HERE = pathlib.Path(__file__).parent

# The text of the README file
README = (HERE / "README.md").read_text()

PKG_NAME = "flyhostel"
version = "1.1.7"

setup(
    name=PKG_NAME,
    version=version,
    packages = find_packages(),
    extras_require={
        "sensor": ["pyserial"],
        "quant": [
            "recordtype",
            "trajectorytools-shaliulab==0.3.5"
        ],
    },
    install_requires=[
        "zeitgeber>=0.0.2",
        "matplotlib",
        "pyaml",
        "imgstore-shaliulab>=0.4.0",
        "pandas",
        "confapp-shaliulab",
        "scikit-learn",
        "recordtype",
        "tqdm",
        "h5py",
        "hdf5storage",
        "yolov7tools==1.1",
        "vidio",
        "webcolors",
        "GitPython",
        # "sleap-io",
        #"cupy>=12.2.0",
        #"cudf>=23.10.02",
    ],
    entry_points={
        "console_scripts": [
            "fh=flyhostel.__main__:main",
            "fh-server=flyhostel.server.server:main",
            "fh-validate=flyhostel.data.bin.dashboard:main",
            "missing-chunk-detector=flyhostel.utils.missing_chunk_detector:main",
            "compile-pose=flyhostel.data.bin.pose:main",
            "train-umap=flyhostel.data.bin.umap:main",
            "project-pose=flyhostel.data.bin.behavior:main",
            "predict-behavior=flyhostel.data.bin.ethogram:main",
            "draw-ethogram=flyhostel.data.bin.ethogram:draw_ethogram",
            "compute-interactions=flyhostel.data.bin.interactions:main",
            "filter-pose=flyhostel.data.bin.filter_pose:main",
            "export-filter-pose=flyhostel.data.bin.export_filter_pose:main",
            "preprocess-pose=flyhostel.data.bin.pose:preprocess",
            "annotate-video=flyhostel.data.bin.movie:main",
            "list-frames-with-no-animals=flyhostel.data.sqlite3.missing_animals:main",
            "make-identogram=flyhostel.data.bin.human_validation.annotate:main",
            "export-images=flyhostel.data.bin.human_validation.export:main",
            "auto-annotate-qc=flyhostel.data.bin.human_validation.qc:main",
            "integrate-human-annotations=flyhostel.data.bin.human_validation.integrate:main",
            "save-human-annotations=flyhostel.data.bin.human_validation.integrate:save",
            "cvat-label-constructor=flyhostel.data.bin.human_validation.label_constructor:main",
            "fh-make-video=flyhostel.data.bin.video:main",
            "fh-make-csv=flyhostel.data.bin.video:save_csv",
            "find-chunk-interval=flyhostel.data.bin.find_chunk_interval:main",
            "get-wavelet-profile=flyhostel.data.bin.utils:main_get_wavelet_profile",
            "get-framerate=flyhostel.data.bin.utils:main_get_framerate",
            "get-chunksize=flyhostel.data.bin.utils:main_get_chunksize",
            "get-number-of-animals=flyhostel.data.bin.utils:main_get_number_of_animals",

            ]
    },
)


warnings.warn("Make sure that idtrackerai, torch, torchvision, confapp, zeitgeber, trajectorytools, feed_integration, dropy are installed")