rwightman
/ triplet_loss.py
Last active
November 21, 2023 10:31
Hacky PyTorch Batch-Hard Triplet Loss and PK samplers
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import torch | |
| from torch import nn | |
| import torch.nn.functional as F | |
| from collections import OrderedDict | |
| import math | |
| def pdist(v): | |
| dist = torch.norm(v[:, None] - v, dim=2, p=2) | |
| return dist |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| def bb_intersection_over_union(boxA, boxB): | |
| # determine the (x, y)-coordinates of the intersection rectangle | |
| xA = max(boxA[0], boxB[0]) | |
| yA = max(boxA[1], boxB[1]) | |
| xB = min(boxA[2], boxB[2]) | |
| yB = min(boxA[3], boxB[3]) | |
| # compute the area of intersection rectangle | |
| interArea = abs(max((xB - xA, 0)) * max((yB - yA), 0)) | |
| if interArea == 0: |
aparrish
/ understanding-word-vectors.ipynb
Last active
July 27, 2024 14:45
Understanding word vectors: A tutorial for "Reading and Writing Electronic Text," a class I teach at ITP. (Python 2.7) Code examples released under CC0 https://creativecommons.org/choose/zero/, other text released under CC BY 4.0 https://creativecommons.org/licenses/by/4.0/
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.