Spaces:
Runtime error
Runtime error
sfmig
commited on
Commit
•
4f1cf17
1
Parent(s):
17145cb
changed requirements
Browse files- app.py +1 -1
- requirements.txt +6 -104
- requirements_OLD.txt +0 -6
app.py
CHANGED
@@ -186,7 +186,7 @@ def predict_animal_mask(im):
|
|
186 |
bboxes = outputs.pred_boxes
|
187 |
masks = outputs.pred_masks # torch.Size([1, 100, 200, 200])
|
188 |
|
189 |
-
# postprocess the
|
190 |
label_per_pixel = torch.argmax(masks.squeeze(),dim=0).detach().numpy()
|
191 |
color_mask = np.zeros(image.size+(3,))
|
192 |
for lbl, color in enumerate(ade_palette()):
|
|
|
186 |
bboxes = outputs.pred_boxes
|
187 |
masks = outputs.pred_masks # torch.Size([1, 100, 200, 200])
|
188 |
|
189 |
+
# postprocess the mask (numpy arrays)
|
190 |
label_per_pixel = torch.argmax(masks.squeeze(),dim=0).detach().numpy()
|
191 |
color_mask = np.zeros(image.size+(3,))
|
192 |
for lbl, color in enumerate(ade_palette()):
|
requirements.txt
CHANGED
@@ -1,104 +1,6 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
async-timeout==4.0.2
|
8 |
-
asynctest==0.13.0
|
9 |
-
attrs==21.4.0
|
10 |
-
backoff==1.10.0
|
11 |
-
bcrypt==3.2.2
|
12 |
-
cachetools==5.2.0
|
13 |
-
certifi @ file:///private/var/folders/sy/f16zz6x50xz3113nwtb9bvq00000gp/T/abs_83242e7e-f82d-4a71-8ef2-9d71d212d249gu_wxmeq/croots/recipe/certifi_1655968827803/work/certifi
|
14 |
-
cffi==1.15.1
|
15 |
-
charset-normalizer==2.1.0
|
16 |
-
click==8.1.3
|
17 |
-
cryptography==37.0.2
|
18 |
-
cycler==0.11.0
|
19 |
-
fastapi==0.78.0
|
20 |
-
ffmpy==0.3.0
|
21 |
-
filelock==3.7.1
|
22 |
-
flatbuffers==1.12
|
23 |
-
fonttools==4.33.3
|
24 |
-
frozenlist==1.3.0
|
25 |
-
fsspec==2022.5.0
|
26 |
-
gast==0.4.0
|
27 |
-
google-auth==2.9.0
|
28 |
-
google-auth-oauthlib==0.4.6
|
29 |
-
google-pasta==0.2.0
|
30 |
-
gradio==3.0.24
|
31 |
-
grpcio==1.47.0
|
32 |
-
h11==0.12.0
|
33 |
-
h5py==3.7.0
|
34 |
-
httpcore==0.15.0
|
35 |
-
httpx==0.23.0
|
36 |
-
huggingface-hub==0.8.1
|
37 |
-
idna==3.3
|
38 |
-
importlib-metadata==4.12.0
|
39 |
-
Jinja2==3.1.2
|
40 |
-
keras==2.9.0
|
41 |
-
Keras-Preprocessing==1.1.2
|
42 |
-
kiwisolver==1.4.3
|
43 |
-
libclang==14.0.1
|
44 |
-
linkify-it-py==1.0.3
|
45 |
-
Markdown==3.3.7
|
46 |
-
markdown-it-py==2.1.0
|
47 |
-
MarkupSafe==2.1.1
|
48 |
-
matplotlib==3.5.2
|
49 |
-
mdit-py-plugins==0.3.0
|
50 |
-
mdurl==0.1.1
|
51 |
-
monotonic==1.6
|
52 |
-
multidict==6.0.2
|
53 |
-
numpy==1.21.6
|
54 |
-
oauthlib==3.2.0
|
55 |
-
olefile==0.46
|
56 |
-
opt-einsum==3.3.0
|
57 |
-
orjson==3.7.6
|
58 |
-
packaging==21.3
|
59 |
-
pandas==1.3.5
|
60 |
-
paramiko==2.11.0
|
61 |
-
Pillow @ file:///Users/runner/miniforge3/conda-bld/pillow_1602493105252/work
|
62 |
-
protobuf==3.19.4
|
63 |
-
pyasn1==0.4.8
|
64 |
-
pyasn1-modules==0.2.8
|
65 |
-
pycparser==2.21
|
66 |
-
pycryptodome==3.15.0
|
67 |
-
pydantic==1.9.1
|
68 |
-
pydub==0.25.1
|
69 |
-
PyNaCl==1.5.0
|
70 |
-
pyparsing==3.0.9
|
71 |
-
python-dateutil==2.8.2
|
72 |
-
python-multipart==0.0.5
|
73 |
-
pytz==2022.1
|
74 |
-
PyYAML==6.0
|
75 |
-
regex==2022.6.2
|
76 |
-
requests==2.28.1
|
77 |
-
requests-oauthlib==1.3.1
|
78 |
-
rfc3986==1.5.0
|
79 |
-
rsa==4.8
|
80 |
-
six==1.16.0
|
81 |
-
sniffio==1.2.0
|
82 |
-
starlette==0.19.1
|
83 |
-
tensorboard==2.9.1
|
84 |
-
tensorboard-data-server==0.6.1
|
85 |
-
tensorboard-plugin-wit==1.8.1
|
86 |
-
tensorflow==2.9.1
|
87 |
-
tensorflow-estimator==2.9.0
|
88 |
-
tensorflow-io-gcs-filesystem==0.26.0
|
89 |
-
termcolor==1.1.0
|
90 |
-
timm==0.5.4
|
91 |
-
tokenizers==0.12.1
|
92 |
-
torch==1.12.0
|
93 |
-
torchaudio==0.12.0
|
94 |
-
torchvision==0.13.0
|
95 |
-
tqdm==4.64.0
|
96 |
-
transformers==4.20.1
|
97 |
-
typing_extensions==4.3.0
|
98 |
-
uc-micro-py==1.0.1
|
99 |
-
urllib3==1.26.9
|
100 |
-
uvicorn==0.18.2
|
101 |
-
Werkzeug==2.1.2
|
102 |
-
wrapt==1.14.1
|
103 |
-
yarl==1.7.2
|
104 |
-
zipp==3.8.0
|
|
|
1 |
+
transformers
|
2 |
+
Pillow
|
3 |
+
gradio
|
4 |
+
numpy
|
5 |
+
torch
|
6 |
+
torchvision
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
requirements_OLD.txt
DELETED
@@ -1,6 +0,0 @@
|
|
1 |
-
transformers
|
2 |
-
Pillow
|
3 |
-
gradio
|
4 |
-
numpy
|
5 |
-
torch
|
6 |
-
torchvision
|
|
|
|
|
|
|
|
|
|
|
|
|
|