Update README.md
Browse files
README.md
CHANGED
@@ -3,8 +3,21 @@ license: cc-by-4.0
|
|
3 |
---
|
4 |
|
5 |
# TACDEC-model
|
6 |
-
This is a simple model with weights
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
|
8 |
## More information
|
9 |
-
For
|
10 |
[TACDEC](https://huggingface.co/datasets/SimulaMet-HOST/TACDEC)
|
|
|
3 |
---
|
4 |
|
5 |
# TACDEC-model
|
6 |
+
This is a simple model with weights and reproducible code for the results in TACDEC-paper.
|
7 |
+
|
8 |
+
What you can find in this repo is:
|
9 |
+
- The simple [model](https://huggingface.co/SimulaMet-HOST/TACDEC-model/resolve/main/model.py?download=true) used in the TACDEC-paper
|
10 |
+
- The [weights](https://huggingface.co/SimulaMet-HOST/TACDEC-model/resolve/main/simple_model_weights.pt?download=true) used in the proof-of-concept section in the TACDEC-paper
|
11 |
+
- A first notebook, [feature_extraction.ipynb](https://huggingface.co/SimulaMet-HOST/TACDEC-model/resolve/main/feature_extraction.ipynb?download=true), that contains a feature extraction process using DINOv2.
|
12 |
+
- A second notebook, [train_classifier.ipynb](https://huggingface.co/SimulaMet-HOST/TACDEC-model/resolve/main/train_classifier.ipynb?download=true), that uses the features that were either extracted using the first notebook, or downloaded directly from (TACDEC repo)[https://huggingface.co/datasets/SimulaMet-HOST/TACDEC].
|
13 |
+
|
14 |
+
We highly recommend downloading the already extracted and concatenated features, together with the concatenated labels. If you just wish to try the dataset/model, you would then just have to run the second notebook.
|
15 |
+
|
16 |
+
|
17 |
+
If you hold more interest in DINOv2, the **feature_extraction.ipynb** could hold good value.
|
18 |
+
|
19 |
+
In both notebooks, there should be good enough documentation, but should you have any questions [TACDEC](https://huggingface.co/datasets/SimulaMet-HOST/TACDEC).
|
20 |
|
21 |
## More information
|
22 |
+
For any other information or information about the dataset:
|
23 |
[TACDEC](https://huggingface.co/datasets/SimulaMet-HOST/TACDEC)
|