add detectron2

pull/139/head
jwyang 2023-10-05 07:55:19 +00:00
parent 40dbcbc358
commit b120d51969
1 changed files with 7 additions and 1 deletions

View File

@ -12,6 +12,7 @@ classifiers = [
"Programming Language :: Python :: 3", "Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License", "License :: OSI Approved :: Apache Software License",
] ]
dependencies = [ dependencies = [
"torch==2.1.0", "torch==2.1.0",
"torchvision==0.16.0", "torchvision==0.16.0",
@ -46,9 +47,14 @@ dependencies = [
"deepspeed==0.10.3", "deepspeed==0.10.3",
"wandb==0.15.12", "wandb==0.15.12",
"infinibatch==0.1.1", "infinibatch==0.1.1",
"gradio==3.42.0" "gradio==3.42.0",
"openai-whisper",
] ]
[tool.poetry.dependencies]
detectron2 = {git = "https://github.com/MaureenZOU/detectron2-xyz.git"}
[project.urls] [project.urls]
"Paper" = "https://arxiv.org/abs/2304.06718" "Paper" = "https://arxiv.org/abs/2304.06718"
"Code" = "https://github.com/UX-Decoder/Segment-Everything-Everywhere-All-At-Once/tree/v1.0" "Code" = "https://github.com/UX-Decoder/Segment-Everything-Everywhere-All-At-Once/tree/v1.0"