Args: classes (list[int] or None): scores (list[float] or None): class_names (list[str] or None): Returns: list[str] or None
(classes, scores, class_names)
| 221 | |
| 222 | |
| 223 | def _create_text_labels(classes, scores, class_names): |
| 224 | """ |
| 225 | Args: |
| 226 | classes (list[int] or None): |
| 227 | scores (list[float] or None): |
| 228 | class_names (list[str] or None): |
| 229 | |
| 230 | Returns: |
| 231 | list[str] or None |
| 232 | """ |
| 233 | labels = None |
| 234 | if classes is not None and class_names is not None and len(class_names) > 0: |
| 235 | labels = [class_names[i] for i in classes] |
| 236 | if scores is not None: |
| 237 | if labels is None: |
| 238 | labels = ["{:.0f}%".format(s * 100) for s in scores] |
| 239 | else: |
| 240 | labels = ["{} {:.0f}%".format(l, s * 100) for l, s in zip(labels, scores)] |
| 241 | return labels |
| 242 | |
| 243 | |
| 244 | class VisImage: |
no outgoing calls
no test coverage detected