ModelsAPI/install_api.py at master · Instinct323/ModelsAPI · 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
from api import Path
from api.utils.package_installer import Package

SOURCE = Path("api")
PACKAGE = Package(
    SOURCE,
    dependency=Path("config/dependency.yaml").yaml(),
    packages=[f.name for f in SOURCE.iterdir() if f.is_dir()]
)

if __name__ == '__main__':
    ROOT = Path("/media/tongzj/Data/Workbench/Lab")

    REQUIRE = {
        ROOT / "GCNGrasp-VP/gcngrasp": [
            "utils", "dataset/task_grasp", "utils/func_invoker", "utils/chat_tool", "utils/bayes_opt",
            "contact_graspnet", "grounded_sam", "depth_anything",
            "utils/camera", "utils/sv_extension", "utils/package_installer",
            "utils/realsense", "utils/msg_tool", "utils/zmq_socket",
            "utils/franka_controller", "utils/curobo_planning"
        ],
        ROOT / "GCNGrasp-VP-custom/api": [
            "utils/object_tracker", "utils/shelf_manager"
        ],
        ROOT / "shelf_manager/api": [
            "utils/object_tracker", "utils/shelf_manager", "utils/sv_extension", "grounded_sam"
        ],
        ROOT / "GSAM-CGN": [
            "contact_graspnet", "grounding_dino", "segment_anything",
            "utils", "utils/camera", "utils/o3d_extension", "utils/pose_tf", "utils/sv_extension"
        ]
    }

    for dst, require in REQUIRE.items():
        PACKAGE.install(dst, require=require)