mirrors
/
GLEE
mirror of https://github.com/FoundationVision/GLEE
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

updata README

pull/18/head
wjf5203 2024-03-21 11:13:59 +08:00
parent 7a021a077b
commit 68c003f61f
3 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
assets/INSTALL.md
View File

@ -7,7 +7,7 @@ We test the codes in the following environments, other versions may also be comp
- Pytorch 2.1.0 - Pytorch 2.1.0
- Torchvison 0.16.0 - Torchvison 0.16.0
## Install environment for UNINEXT ## Install environment for GLEE
``` ```
pip3 install shapely==1.7.1 pip3 install shapely==1.7.1

BIN
assets/images/imagetask.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 636 KiB

After

Width:  |  Height:  |  Size: 802 KiB

2
projects/GLEE/glee/GLEE.py
View File

@ -380,7 +380,7 @@ class GLEE(nn.Module):
self.inference_rvos(batched_inputs, prompt_list, task) self.inference_rvos(batched_inputs, prompt_list, task)
return return
elif task in ['ytbvos']: elif task in ['ytbvos']:
self.inference_ytbvos_uninext(batched_inputs, prompt_list, task) self.inference_ytbvos(batched_inputs, prompt_list, task)
elif task in ['omnilabel']: elif task in ['omnilabel']:
return self.omnilabel_inference(batched_inputs, task) return self.omnilabel_inference(batched_inputs, task)
else: else: