Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
`torch` `torchvision` `tqdm`
|
2 |
|
3 |
This implementation is inspired by **"Prototypical Networks for Few-Shot Learning" (Snell et al., 2017)**.
|
@@ -65,4 +78,4 @@ Prototypical Networks are a powerful approach for **few-shot learning**, where t
|
|
65 |
* **Embedding Representation with CNN**: Each input image is passed through a convolutional encoder to obtain a feature embedding.
|
66 |
* **Prototype Computation**: The prototype for each class is computed as the mean of the embeddings of support samples belonging to that class.
|
67 |
* **Distance-Based Classification**: Query samples are classified based on the distance (using `torch.cdist`) to the nearest prototype.
|
68 |
-
* **Optimization**: The network is trained to minimize the distance between query samples and their correct prototypes while maximizing the distance to incorrect ones.
|
|
|
1 |
+
---
|
2 |
+
license: mit
|
3 |
+
datasets:
|
4 |
+
- GATE-engine/omniglot
|
5 |
+
language:
|
6 |
+
- ko
|
7 |
+
pipeline_tag: image-classification
|
8 |
+
tags:
|
9 |
+
- pytorch
|
10 |
+
- few-shot-learning
|
11 |
+
- one-shot-learning
|
12 |
+
- meta-learning
|
13 |
+
---
|
14 |
`torch` `torchvision` `tqdm`
|
15 |
|
16 |
This implementation is inspired by **"Prototypical Networks for Few-Shot Learning" (Snell et al., 2017)**.
|
|
|
78 |
* **Embedding Representation with CNN**: Each input image is passed through a convolutional encoder to obtain a feature embedding.
|
79 |
* **Prototype Computation**: The prototype for each class is computed as the mean of the embeddings of support samples belonging to that class.
|
80 |
* **Distance-Based Classification**: Query samples are classified based on the distance (using `torch.cdist`) to the nearest prototype.
|
81 |
+
* **Optimization**: The network is trained to minimize the distance between query samples and their correct prototypes while maximizing the distance to incorrect ones.
|