



import json
import os
from functools import lru_cache
from typing import TYPE_CHECKING, List, Optional, Tuple, Union

import numpy as np
import regex as re

from ...utils import is_tf_available, is_torch_available, logging


if TYPE_CHECKING:
	if is_torch_available():
		import torch
	if is_tf_available():
		import tensorflow as tf

from ...tokenization_utils import AddedToken, PreTrainedTokenizer


A__       =			logging.get_logger(__name__)

A__       =			{
    """vocab_file""": """vocab.json""",
    """merges_file""": """merges.txt""",
}

A__       =			{
    """vocab_file""": {
        """Salesforce/codegen-350M-mono""": """https://huggingface.co/Salesforce/codegen-350M-mono/resolve/main/vocab.json""",
    },
    """merges_file""": {
        """Salesforce/codegen-350M-mono""": """https://huggingface.co/Salesforce/codegen-350M-mono/resolve/main/merges.txt""",
    },
}

A__       =			{
    """Salesforce/codegen-350M-mono""": 20_48,
}


@lru_cache()
def 						_UpperCAmelCase							( ):





					"""simple docstring"""

					_lowerCAmelCase         =       (
					    list(range(ord("""!"""     )      ,		ord("""~"""     ) + 1     )     ) + list(range(ord("""¡"""     )      ,		ord("""¬"""     ) + 1     )     ) + list(range(ord("""®"""     )      ,		ord("""ÿ"""     ) + 1     )     )
					)
					_lowerCAmelCase         =       bs[:]
					_lowerCAmelCase         =       0
					for b in range(2**8     ):
										if b not in bs:
															bs.append(A_     )
															cs.append(2**8 + n     )
															n += 1
					_lowerCAmelCase         =       [chr(A_     ) for n in cs]
					return dict(zip(A_      ,		A_     )     )


def 						_UpperCAmelCase							( snake_case     ):





					"""simple docstring"""

					_lowerCAmelCase         =       set()
					_lowerCAmelCase         =       word[0]
					for char in word[1:]:
										pairs.add((prev_char, char)     )
										_lowerCAmelCase         =       char
					return pairs





class    __lowerCAmelCase					(	lowerCamelCase__			):
				__lowerCamelCase					=							VOCAB_FILES_NAMES
				__lowerCamelCase					=							PRETRAINED_VOCAB_FILES_MAP
				__lowerCamelCase					=							PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
				__lowerCamelCase					=							['''input_ids''', '''attention_mask''']




				def __init__(				self		,							_snake_case		,							_snake_case		,							_snake_case="replace"		,							_snake_case="<|endoftext|>"		,							_snake_case="<|endoftext|>"		,							_snake_case="<|endoftext|>"		,							_snake_case=None		,							_snake_case=False		,							_snake_case=False		,							**_snake_case		,							):



									"""simple docstring"""





									_lowerCAmelCase         =       AddedToken(SCREAMING_SNAKE_CASE_		,							lstrip=SCREAMING_SNAKE_CASE_		,							rstrip=SCREAMING_SNAKE_CASE_						) if isinstance(SCREAMING_SNAKE_CASE_		,							SCREAMING_SNAKE_CASE_						) else bos_token
									_lowerCAmelCase         =       AddedToken(SCREAMING_SNAKE_CASE_		,							lstrip=SCREAMING_SNAKE_CASE_		,							rstrip=SCREAMING_SNAKE_CASE_						) if isinstance(SCREAMING_SNAKE_CASE_		,							SCREAMING_SNAKE_CASE_						) else eos_token
									_lowerCAmelCase         =       AddedToken(SCREAMING_SNAKE_CASE_		,							lstrip=SCREAMING_SNAKE_CASE_		,							rstrip=SCREAMING_SNAKE_CASE_						) if isinstance(SCREAMING_SNAKE_CASE_		,							SCREAMING_SNAKE_CASE_						) else unk_token
									_lowerCAmelCase         =       AddedToken(SCREAMING_SNAKE_CASE_		,							lstrip=SCREAMING_SNAKE_CASE_		,							rstrip=SCREAMING_SNAKE_CASE_						) if isinstance(SCREAMING_SNAKE_CASE_		,							SCREAMING_SNAKE_CASE_						) else pad_token
									super().__init__(
									    errors=SCREAMING_SNAKE_CASE_		,							unk_token=SCREAMING_SNAKE_CASE_		,							bos_token=SCREAMING_SNAKE_CASE_		,							eos_token=SCREAMING_SNAKE_CASE_		,							pad_token=SCREAMING_SNAKE_CASE_		,							add_prefix_space=SCREAMING_SNAKE_CASE_		,							add_bos_token=SCREAMING_SNAKE_CASE_		,							**SCREAMING_SNAKE_CASE_		,							)
									_lowerCAmelCase         =       add_bos_token

									with open(SCREAMING_SNAKE_CASE_		,							encoding="""utf-8"""						) as vocab_handle:
														_lowerCAmelCase         =       json.load(SCREAMING_SNAKE_CASE_						)
									_lowerCAmelCase         =       {v: k for k, v in self.encoder.items()}
									_lowerCAmelCase         =       errors  # how to handle errors in decoding
									_lowerCAmelCase         =       bytes_to_unicode()
									_lowerCAmelCase         =       {v: k for k, v in self.byte_encoder.items()}
									with open(SCREAMING_SNAKE_CASE_		,							encoding="""utf-8"""						) as merges_handle:
														_lowerCAmelCase         =       merges_handle.read().split("""\n"""						)[1:-1]
									_lowerCAmelCase         =       [tuple(merge.split()						) for merge in bpe_merges]
									_lowerCAmelCase         =       dict(zip(SCREAMING_SNAKE_CASE_		,							range(len(SCREAMING_SNAKE_CASE_						)						)						)						)
									_lowerCAmelCase         =       {}
									_lowerCAmelCase         =       add_prefix_space

									# Should have added re.IGNORECASE so BPE merges can happen for capitalized versions of contractions
									_lowerCAmelCase         =       re.compile(R"""\'s|\'t|\'re|\'ve|\'m|\'ll|\'d| ?\p{L}+| ?\p{N}+| ?[^\s\p{L}\p{N}]+|\s+(?!\S)|\s+"""						)




				@property
				def 				snake_case  (				self						):



									"""simple docstring"""





									return len(self.encoder						)




				def 				snake_case  (				self						):



									"""simple docstring"""





									return dict(self.encoder		,							**self.added_tokens_encoder						)




				def 				snake_case  (				self		,							_snake_case						):



									"""simple docstring"""





									if token in self.cache:
														return self.cache[token]
									_lowerCAmelCase         =       tuple(SCREAMING_SNAKE_CASE_						)
									_lowerCAmelCase         =       get_pairs(SCREAMING_SNAKE_CASE_						)

									if not pairs:
														return token

									while True:
														_lowerCAmelCase         =       min(SCREAMING_SNAKE_CASE_		,							key=lambda _snake_case						: self.bpe_ranks.get(SCREAMING_SNAKE_CASE_		,							float("""inf"""						)						)						)
														if bigram not in self.bpe_ranks:
																			break
														_lowerCAmelCase         =       bigram
														_lowerCAmelCase         =       []
														_lowerCAmelCase         =       0
														while i < len(SCREAMING_SNAKE_CASE_						):
																			try:
																								_lowerCAmelCase         =       word.index(SCREAMING_SNAKE_CASE_		,							SCREAMING_SNAKE_CASE_						)
																			except ValueError:
																								new_word.extend(word[i:]						)
																								break
																			else:
																								new_word.extend(word[i:j]						)
																								_lowerCAmelCase         =       j

																			if word[i] == first and i < len(SCREAMING_SNAKE_CASE_						) - 1 and word[i + 1] == second:
																								new_word.append(first + second						)
																								i += 2
																			else:
																								new_word.append(word[i]						)
																								i += 1
														_lowerCAmelCase         =       tuple(SCREAMING_SNAKE_CASE_						)
														_lowerCAmelCase         =       new_word
														if len(SCREAMING_SNAKE_CASE_						) == 1:
																			break
														else:
																			_lowerCAmelCase         =       get_pairs(SCREAMING_SNAKE_CASE_						)
									_lowerCAmelCase         =       ''' '''.join(SCREAMING_SNAKE_CASE_						)
									_lowerCAmelCase         =       word
									return word




				def 				snake_case  (				self		,							_snake_case		,							_snake_case=None						):



									"""simple docstring"""





									if self.add_bos_token:
														_lowerCAmelCase         =       [self.bos_token_id]
									else:
														_lowerCAmelCase         =       []

									_lowerCAmelCase         =       bos_token_ids + token_ids_a

									if token_ids_a is None:
														return output

									return output + bos_token_ids + token_ids_a




				def 				snake_case  (				self		,							_snake_case						):



									"""simple docstring"""





									_lowerCAmelCase         =       []
									for token in re.findall(self.pat		,							SCREAMING_SNAKE_CASE_						):
														_lowerCAmelCase         =       ''''''.join(
														    self.byte_encoder[b] for b in token.encode("""utf-8"""						)						)  # Maps all our bytes to unicode strings, avoiding control tokens of the BPE (spaces in our case)
														bpe_tokens.extend(bpe_token for bpe_token in self.bpe(SCREAMING_SNAKE_CASE_						).split(""" """						)						)
									return bpe_tokens




				def 				snake_case  (				self		,							_snake_case						):



									"""simple docstring"""





									return self.encoder.get(SCREAMING_SNAKE_CASE_		,							self.encoder.get(self.unk_token						)						)




				def 				snake_case  (				self		,							_snake_case						):



									"""simple docstring"""





									return self.decoder.get(SCREAMING_SNAKE_CASE_						)




				def 				snake_case  (				self		,							_snake_case						):



									"""simple docstring"""





									_lowerCAmelCase         =       ''''''.join(SCREAMING_SNAKE_CASE_						)
									_lowerCAmelCase         =       bytearray([self.byte_decoder[c] for c in text]						).decode("""utf-8"""		,							errors=self.errors						)
									return text




				def 				snake_case  (				self		,							_snake_case		,							_snake_case = None						):



									"""simple docstring"""





									if not os.path.isdir(SCREAMING_SNAKE_CASE_						):
														logger.error(F'Vocabulary path ({save_directory}) should be a directory'						)
														return
									_lowerCAmelCase         =       os.path.join(
									    SCREAMING_SNAKE_CASE_		,							(filename_prefix + """-""" if filename_prefix else """""") + VOCAB_FILES_NAMES["""vocab_file"""]						)
									_lowerCAmelCase         =       os.path.join(
									    SCREAMING_SNAKE_CASE_		,							(filename_prefix + """-""" if filename_prefix else """""") + VOCAB_FILES_NAMES["""merges_file"""]						)

									with open(SCREAMING_SNAKE_CASE_		,							"""w"""		,							encoding="""utf-8"""						) as f:
														f.write(json.dumps(self.encoder		,							indent=2		,							sort_keys=SCREAMING_SNAKE_CASE_		,							ensure_ascii=SCREAMING_SNAKE_CASE_						) + """\n"""						)

									_lowerCAmelCase         =       0
									with open(SCREAMING_SNAKE_CASE_		,							"""w"""		,							encoding="""utf-8"""						) as writer:
														writer.write("""#version: 0.2\n"""						)
														for bpe_tokens, token_index in sorted(self.bpe_ranks.items()		,							key=lambda _snake_case						: kv[1]						):
																			if index != token_index:
																								logger.warning(
																								    F'Saving vocabulary to {merge_file}: BPE merge indices are not consecutive.'
																								    """ Please check that the tokenizer is not corrupted!"""						)
																								_lowerCAmelCase         =       token_index
																			writer.write(""" """.join(SCREAMING_SNAKE_CASE_						) + """\n"""						)
																			index += 1

									return vocab_file, merge_file




				def 				snake_case  (				self		,							_snake_case		,							_snake_case=False		,							**_snake_case						):



									"""simple docstring"""





									_lowerCAmelCase         =       kwargs.pop("""add_prefix_space"""		,							self.add_prefix_space						)
									if is_split_into_words or add_prefix_space:
														_lowerCAmelCase         =       ''' ''' + text
									return (text, kwargs)




				def 				snake_case  (				self		,							_snake_case		,							_snake_case = False		,							_snake_case = None		,							_snake_case = None		,							**_snake_case		,							):



									"""simple docstring"""





									_lowerCAmelCase         =       super()._decode(
									    token_ids=SCREAMING_SNAKE_CASE_		,							skip_special_tokens=SCREAMING_SNAKE_CASE_		,							clean_up_tokenization_spaces=SCREAMING_SNAKE_CASE_		,							**SCREAMING_SNAKE_CASE_		,							)

									if truncate_before_pattern is not None and len(SCREAMING_SNAKE_CASE_						) > 0:
														_lowerCAmelCase         =       self.truncate(SCREAMING_SNAKE_CASE_		,							SCREAMING_SNAKE_CASE_						)

									return decoded_text




				def 				snake_case  (				self		,							_snake_case		,							_snake_case						):



									"""simple docstring"""





									def find_re(_snake_case		,							_snake_case		,							_snake_case						):
														_lowerCAmelCase         =       pattern.search(SCREAMING_SNAKE_CASE_		,							SCREAMING_SNAKE_CASE_						)
														return m.start() if m else -1

									_lowerCAmelCase         =       [re.compile(SCREAMING_SNAKE_CASE_		,							re.MULTILINE						) for pattern in truncate_before_pattern]

									_lowerCAmelCase         =       list(re.finditer("""^print"""		,							SCREAMING_SNAKE_CASE_		,							re.MULTILINE						)						)

									if len(SCREAMING_SNAKE_CASE_						) > 1:
														_lowerCAmelCase         =       completion[: prints[1].start()]

									_lowerCAmelCase         =       list(re.finditer("""^def"""		,							SCREAMING_SNAKE_CASE_		,							re.MULTILINE						)						)

									if len(SCREAMING_SNAKE_CASE_						) > 1:
														_lowerCAmelCase         =       completion[: defs[1].start()]

									_lowerCAmelCase         =       0

									_lowerCAmelCase         =       [
									    pos for pos in [find_re(SCREAMING_SNAKE_CASE_		,							SCREAMING_SNAKE_CASE_		,							SCREAMING_SNAKE_CASE_						) for terminal in terminals] if pos != -1
									]

									if len(SCREAMING_SNAKE_CASE_						) > 0:
														return completion[: min(SCREAMING_SNAKE_CASE_						)]
									else:
														return completion





'''simple docstring'''





import colorsys
import io

import cva
import matplotlib as mpl
import matplotlib.colors as mplc
import matplotlib.figure as mplfigure
import numpy as np
import torch
from matplotlib.backends.backend_agg import FigureCanvasAgg

from utils import img_tensorize


UpperCAmelCase            = 1_000

class     a :




				def __init__(				self						:					Optional[int],     SCREAMING_SNAKE_CASE_						:					Tuple,     SCREAMING_SNAKE_CASE_						:					int=1.2,     SCREAMING_SNAKE_CASE_						:					Optional[Any]="g",     SCREAMING_SNAKE_CASE_						:					Any=0.5,     SCREAMING_SNAKE_CASE_						:					int="-",     SCREAMING_SNAKE_CASE_						:					Dict="test_out.jpg",     SCREAMING_SNAKE_CASE_						:					Any=True,     SCREAMING_SNAKE_CASE_						:					Tuple=False,     SCREAMING_SNAKE_CASE_						:					Any=None,     SCREAMING_SNAKE_CASE_						:					str=None,     SCREAMING_SNAKE_CASE_						:					str=0.7,     ):
										if isinstance(SCREAMING_SNAKE_CASE_,     torch.Tensor   ):
																snake_case							:  Dict			=     img.numpy().astype('''np.uint8'''   )
										if isinstance(SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_   ):
																snake_case							:  List[Any]			=     img_tensorize(SCREAMING_SNAKE_CASE_   )
										assert isinstance(SCREAMING_SNAKE_CASE_,     np.ndarray   )

										snake_case,       snake_case							:  Optional[Any]			=     img.shape[1], img.shape[0]
										snake_case							:  int			=     mplfigure.Figure(frameon=SCREAMING_SNAKE_CASE_   )
										snake_case							:  int			=     fig.get_dpi()
										snake_case							:  Optional[int]			=     (width * scale + 1e-2) / dpi
										snake_case							:  List[Any]			=     (height * scale + 1e-2) / dpi
										fig.set_size_inches(SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_   )
										snake_case							:  Any			=     fig.add_axes([0.0, 0.0, 1.0, 1.0]   )
										ax.axis('''off'''   )
										ax.set_xlim(0.0,     SCREAMING_SNAKE_CASE_   )
										ax.set_ylim(SCREAMING_SNAKE_CASE_   )

										snake_case							:  Union[str, Any]			=     saveas
										snake_case							:  str			=     rgb
										snake_case							:  Tuple			=     pynb
										snake_case							:  Optional[int]			=     img
										snake_case							:  List[str]			=     edgecolor
										snake_case							:  str			=     0.5
										snake_case							:  Optional[Any]			=     linestyle
										snake_case							:  Optional[Any]			=     int(np.sqrt(min(SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_   )   ) * scale // 3   )
										snake_case							:  Any			=     width
										snake_case							:  List[str]			=     height
										snake_case							:  Optional[Any]			=     scale
										snake_case							:  int			=     fig
										snake_case							:  Optional[int]			=     ax
										snake_case							:  List[str]			=     pad
										snake_case							:  Tuple			=     idaobj
										snake_case							:  Tuple			=     idaattr
										snake_case							:  Dict			=     FigureCanvasAgg(SCREAMING_SNAKE_CASE_   )




				def 						__snake_case				(				self						:					Tuple,     SCREAMING_SNAKE_CASE_						:					Union[str, Any],     SCREAMING_SNAKE_CASE_						:					int=None   ):
										if color is None:
																snake_case							:  int			=     self.edgecolor
										((snake_case),       (snake_case),       (snake_case),       (snake_case))							:  Optional[int]			=     box
										snake_case							:  List[Any]			=     xa - xa
										snake_case							:  Optional[int]			=     ya - ya
										self.ax.add_patch(
										    mpl.patches.Rectangle(
										        (xa, ya),     SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_,     fill=SCREAMING_SNAKE_CASE_,     edgecolor=SCREAMING_SNAKE_CASE_,     linewidth=self.font_size // 3,     alpha=self.alpha,     linestyle=self.linestyle,     )   )




				def 						__snake_case				(				self						:					Tuple,     SCREAMING_SNAKE_CASE_						:					Optional[Any],     SCREAMING_SNAKE_CASE_						:					str=None,     SCREAMING_SNAKE_CASE_						:					Union[str, Any]=None,     SCREAMING_SNAKE_CASE_						:					Optional[int]=None,     SCREAMING_SNAKE_CASE_						:					Tuple=None   ):
										if len(boxes.shape   ) > 2:
																snake_case							:  Dict			=     boxes[0]
										if len(obj_ids.shape   ) > 1:
																snake_case							:  Optional[int]			=     obj_ids[0]
										if len(obj_scores.shape   ) > 1:
																snake_case							:  str			=     obj_scores[0]
										if len(attr_ids.shape   ) > 1:
																snake_case							:  int			=     attr_ids[0]
										if len(attr_scores.shape   ) > 1:
																snake_case							:  Union[str, Any]			=     attr_scores[0]
										if isinstance(SCREAMING_SNAKE_CASE_,     torch.Tensor   ):
																snake_case							:  List[Any]			=     boxes.numpy()
										if isinstance(SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_   ):
																snake_case							:  str			=     np.array(SCREAMING_SNAKE_CASE_   )
										assert isinstance(SCREAMING_SNAKE_CASE_,     np.ndarray   )
										snake_case							:  Optional[int]			=     np.prod(boxes[:, 2:] - boxes[:, :2],     axis=1   )
										snake_case							:  str			=     np.argsort(-areas   ).tolist()
										snake_case							:  Union[str, Any]			=     boxes[sorted_idxs] if boxes is not None else None
										snake_case							:  Dict			=     obj_ids[sorted_idxs] if obj_ids is not None else None
										snake_case							:  Tuple			=     obj_scores[sorted_idxs] if obj_scores is not None else None
										snake_case							:  List[str]			=     attr_ids[sorted_idxs] if attr_ids is not None else None
										snake_case							:  Optional[int]			=     attr_scores[sorted_idxs] if attr_scores is not None else None

										snake_case							:  Union[str, Any]			=     [self._random_color(maximum=1   ) for _ in range(len(SCREAMING_SNAKE_CASE_   )   )]
										snake_case							:  List[Any]			=     [assigned_colors[idx] for idx in sorted_idxs]
										if obj_ids is not None:
																snake_case							:  Dict			=     self._create_text_labels_attr(SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_   )
																for i in range(len(SCREAMING_SNAKE_CASE_   )   ):
																						snake_case							:  Any			=     assigned_colors[i]
																						self.add_box(boxes[i],     SCREAMING_SNAKE_CASE_   )
																						self.draw_labels(labels[i],     boxes[i],     SCREAMING_SNAKE_CASE_   )




				def 						__snake_case				(				self						:					List[str],     SCREAMING_SNAKE_CASE_						:					int,     SCREAMING_SNAKE_CASE_						:					Optional[int],     SCREAMING_SNAKE_CASE_						:					Union[str, Any]   ):
										snake_case,       snake_case,       snake_case,       snake_case							:  Dict			=     box
										snake_case							:  Union[str, Any]			=     (xa, ya)
										snake_case							:  Any			=     (ya - ya) * (xa - xa)
										snake_case							:  Any			=     _SMALL_OBJ * self.scale
										if instance_area < small or ya - ya < 40 * self.scale:
																if ya >= self.height - 5:
																						snake_case							:  Union[str, Any]			=     (xa, ya)
																else:
																						snake_case							:  Dict			=     (xa, ya)

										snake_case							:  List[str]			=     (ya - ya) / np.sqrt(self.height * self.width   )
										snake_case							:  Union[str, Any]			=     self._change_color_brightness(SCREAMING_SNAKE_CASE_,     brightness_factor=0.7   )
										snake_case							:  int			=     np.clip((height_ratio - 0.02) / 0.08 + 1,     1.2,     2   )
										font_size *= 0.75 * self.font_size

										self.draw_text(
										    text=SCREAMING_SNAKE_CASE_,     position=SCREAMING_SNAKE_CASE_,     color=SCREAMING_SNAKE_CASE_,     )




				def 						__snake_case				(				self						:					Optional[Any],     SCREAMING_SNAKE_CASE_						:					Union[str, Any],     SCREAMING_SNAKE_CASE_						:					Tuple,     SCREAMING_SNAKE_CASE_						:					List[str]="g",     SCREAMING_SNAKE_CASE_						:					List[str]="left",     ):
										snake_case							:  Optional[int]			=     0
										snake_case							:  Tuple			=     self.font_size
										snake_case							:  Tuple			=     np.maximum(list(mplc.to_rgb(SCREAMING_SNAKE_CASE_   )   ),     0.2   )
										snake_case							:  Dict			=     max(0.8,     np.max(SCREAMING_SNAKE_CASE_   )   )
										snake_case							:  List[Any]			=     {
										    '''facecolor''': '''black''',
										    '''alpha''': self.alpha,
										    '''pad''': self.pad,
										    '''edgecolor''': '''none''',
										}
										snake_case,       snake_case							:  str			=     position
										self.ax.text(
										    SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_,     size=font_size * self.scale,     family='''sans-serif''',     bbox=SCREAMING_SNAKE_CASE_,     verticalalignment='''top''',     horizontalalignment=SCREAMING_SNAKE_CASE_,     color=SCREAMING_SNAKE_CASE_,     zorder=10,     rotation=SCREAMING_SNAKE_CASE_,     )




				def 						__snake_case				(				self						:					Any,     SCREAMING_SNAKE_CASE_						:					Any=None   ):
										if saveas is None:
																snake_case							:  List[str]			=     self.saveas
										if saveas.lower().endswith('''.jpg'''   ) or saveas.lower().endswith('''.png'''   ):
																cva.imwrite(
																    SCREAMING_SNAKE_CASE_,     self._get_buffer()[:, :, ::-1],     )
										else:
																self.fig.savefig(SCREAMING_SNAKE_CASE_   )




				def 						__snake_case				(				self						:					Tuple,     SCREAMING_SNAKE_CASE_						:					Optional[int],     SCREAMING_SNAKE_CASE_						:					str,     SCREAMING_SNAKE_CASE_						:					str,     SCREAMING_SNAKE_CASE_						:					int   ):
										snake_case							:  int			=     [self.idaobj[i] for i in classes]
										snake_case							:  str			=     [self.idaattr[i] for i in attr_classes]
										snake_case							:  Optional[Any]			=     [
										    F"""{label} {score:.2f} {attr} {attr_score:.2f}"""
										    for label, score, attr, attr_score in zip(SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_   )
										]
										return labels




				def 						__snake_case				(				self						:					Any,     SCREAMING_SNAKE_CASE_						:					List[Any],     SCREAMING_SNAKE_CASE_						:					List[str]   ):
										snake_case							:  Union[str, Any]			=     [self.idaobj[i] for i in classes]
										if scores is not None:
																if labels is None:
																						snake_case							:  Optional[Any]			=     ['''{:.0f}%'''.format(s * 1_00   ) for s in scores]
																else:
																						snake_case							:  str			=     ['''{} {:.0f}%'''.format(SCREAMING_SNAKE_CASE_,     s * 1_00   ) for li, s in zip(SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_   )]
										return labels




				def 						__snake_case				(				self						:					Union[str, Any],     SCREAMING_SNAKE_CASE_						:					int=2_55   ):
										snake_case							:  Union[str, Any]			=     np.random.randint(0,     len(_COLORS   )   )
										snake_case							:  Optional[int]			=     _COLORS[idx] * maximum
										if not self.rgb:
																snake_case							:  str			=     ret[::-1]
										return ret




				def 						__snake_case				(				self						:					List[Any]   ):
										if not self.pynb:
																snake_case,       snake_case							:  int			=     self.canvas.print_to_buffer()
																if (width, height) != (self.width, self.height):
																						snake_case							:  List[Any]			=     cva.resize(self.img,     (width, height)   )
																else:
																						snake_case							:  Tuple			=     self.img
										else:
																snake_case							:  List[str]			=     io.BytesIO()  # works for cairo backend
																self.canvas.print_rgba(SCREAMING_SNAKE_CASE_   )
																snake_case,       snake_case							:  Union[str, Any]			=     self.width, self.height
																snake_case							:  Optional[Any]			=     buf.getvalue()
																snake_case							:  List[Any]			=     self.img

										snake_case							:  Union[str, Any]			=     np.frombuffer(SCREAMING_SNAKE_CASE_,     dtype='''uint8'''   )
										snake_case							:  Union[str, Any]			=     buffer.reshape(SCREAMING_SNAKE_CASE_,     SCREAMING_SNAKE_CASE_,     4   )
										snake_case,       snake_case							:  Any			=     np.split(SCREAMING_SNAKE_CASE_,     [3],     axis=2   )

										try:
																import numexpr as ne  # fuse them with numexpr

																snake_case							:  str			=     ne.evaluate('''img * (1 - alpha / 255.0) + rgb * (alpha / 255.0)'''   )
										except ImportError:
																snake_case							:  Dict			=     alpha.astype('''float32'''   ) / 255.0
																snake_case							:  Tuple			=     img * (1 - alpha) + rgb * alpha

										return visualized_image.astype('''uint8'''   )




				def 						__snake_case				(				self						:					List[str],     SCREAMING_SNAKE_CASE_						:					Optional[int],     SCREAMING_SNAKE_CASE_						:					Any   ):
										assert brightness_factor >= -1.0 and brightness_factor <= 1.0
										snake_case							:  Any			=     mplc.to_rgb(SCREAMING_SNAKE_CASE_   )
										snake_case							:  List[str]			=     colorsys.rgb_to_hls(*mplc.to_rgb(SCREAMING_SNAKE_CASE_   )   )
										snake_case							:  Union[str, Any]			=     polygon_color[1] + (brightness_factor * polygon_color[1])
										snake_case							:  List[Any]			=     0.0 if modified_lightness < 0.0 else modified_lightness
										snake_case							:  Dict			=     1.0 if modified_lightness > 1.0 else modified_lightness
										snake_case							:  Dict			=     colorsys.hls_to_rgb(polygon_color[0],     SCREAMING_SNAKE_CASE_,     polygon_color[2]   )
										return modified_color




UpperCAmelCase            = (
    np.array(
        [
            0.000,
            0.447,
            0.741,
            0.850,
            0.325,
            0.098,
            0.929,
            0.694,
            0.125,
            0.494,
            0.184,
            0.556,
            0.466,
            0.674,
            0.188,
            0.301,
            0.745,
            0.933,
            0.635,
            0.078,
            0.184,
            0.300,
            0.300,
            0.300,
            0.600,
            0.600,
            0.600,
            1.000,
            0.000,
            0.000,
            1.000,
            0.500,
            0.000,
            0.749,
            0.749,
            0.000,
            0.000,
            1.000,
            0.000,
            0.000,
            0.000,
            1.000,
            0.667,
            0.000,
            1.000,
            0.333,
            0.333,
            0.000,
            0.333,
            0.667,
            0.000,
            0.333,
            1.000,
            0.000,
            0.667,
            0.333,
            0.000,
            0.667,
            0.667,
            0.000,
            0.667,
            1.000,
            0.000,
            1.000,
            0.333,
            0.000,
            1.000,
            0.667,
            0.000,
            1.000,
            1.000,
            0.000,
            0.000,
            0.333,
            0.500,
            0.000,
            0.667,
            0.500,
            0.000,
            1.000,
            0.500,
            0.333,
            0.000,
            0.500,
            0.333,
            0.333,
            0.500,
            0.333,
            0.667,
            0.500,
            0.333,
            1.000,
            0.500,
            0.667,
            0.000,
            0.500,
            0.667,
            0.333,
            0.500,
            0.667,
            0.667,
            0.500,
            0.667,
            1.000,
            0.500,
            1.000,
            0.000,
            0.500,
            1.000,
            0.333,
            0.500,
            1.000,
            0.667,
            0.500,
            1.000,
            1.000,
            0.500,
            0.000,
            0.333,
            1.000,
            0.000,
            0.667,
            1.000,
            0.000,
            1.000,
            1.000,
            0.333,
            0.000,
            1.000,
            0.333,
            0.333,
            1.000,
            0.333,
            0.667,
            1.000,
            0.333,
            1.000,
            1.000,
            0.667,
            0.000,
            1.000,
            0.667,
            0.333,
            1.000,
            0.667,
            0.667,
            1.000,
            0.667,
            1.000,
            1.000,
            1.000,
            0.000,
            1.000,
            1.000,
            0.333,
            1.000,
            1.000,
            0.667,
            1.000,
            0.333,
            0.000,
            0.000,
            0.500,
            0.000,
            0.000,
            0.667,
            0.000,
            0.000,
            0.833,
            0.000,
            0.000,
            1.000,
            0.000,
            0.000,
            0.000,
            0.167,
            0.000,
            0.000,
            0.333,
            0.000,
            0.000,
            0.500,
            0.000,
            0.000,
            0.667,
            0.000,
            0.000,
            0.833,
            0.000,
            0.000,
            1.000,
            0.000,
            0.000,
            0.000,
            0.167,
            0.000,
            0.000,
            0.333,
            0.000,
            0.000,
            0.500,
            0.000,
            0.000,
            0.667,
            0.000,
            0.000,
            0.833,
            0.000,
            0.000,
            1.000,
            0.000,
            0.000,
            0.000,
            0.143,
            0.143,
            0.143,
            0.857,
            0.857,
            0.857,
            1.000,
            1.000,
            1.000,
        ]
    )
    .astype(np.floataa)
    .reshape(-1, 3)
)

