mirrors
/
Segment-Everything-Everywhere-All-At-Once
mirror of https://github.com/UX-Decoder/Segment-Everything-Everywhere-All-At-Once.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix nltk package

pull/139/head
jwyang 2023-10-05 17:57:17 +00:00
parent 7f21a4fff9
commit 4e58f99806
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modeling/language/misc.py
View File

@ -2,11 +2,12 @@ import random
import torch import torch
import nltk import nltk
nltk.data.path.append('/mnt/data/nltk_data')
import numpy as np import numpy as np
from utils.constants import IMAGENET_DEFAULT_TEMPLATES from utils.constants import IMAGENET_DEFAULT_TEMPLATES
nltk.download('punkt', quiet=True)
nltk.download('averaged_perceptron_tagger', quiet=True)
def get_tag(tokenized, tags): def get_tag(tokenized, tags):
if not isinstance(tags, (list, tuple)): if not isinstance(tags, (list, tuple)):