Commit
·
9e6d6a7
1
Parent(s):
af7f33f
Amend the JAX model config file: add image size.
Browse files- README.md +4 -0
- sw_jax_cv_config.json +2 -1
README.md
CHANGED
|
@@ -20,6 +20,10 @@ Tags with less than 600 images were filtered out.
|
|
| 20 |
`P=R: threshold = 0.2521, F1 = 0.4411`
|
| 21 |
|
| 22 |
## What's new
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
Model v1.0/Dataset v3:
|
| 24 |
More training images, more and up-to-date tags (up to 2024-02-28).
|
| 25 |
Now `timm` compatible! Load it up and give it a spin using the canonical one-liner!
|
|
|
|
| 20 |
`P=R: threshold = 0.2521, F1 = 0.4411`
|
| 21 |
|
| 22 |
## What's new
|
| 23 |
+
Model v1.1/Dataset v3:
|
| 24 |
+
Amended the JAX model config file: add image size.
|
| 25 |
+
No change to the trained weights.
|
| 26 |
+
|
| 27 |
Model v1.0/Dataset v3:
|
| 28 |
More training images, more and up-to-date tags (up to 2024-02-28).
|
| 29 |
Now `timm` compatible! Load it up and give it a spin using the canonical one-liner!
|
sw_jax_cv_config.json
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
{
|
|
|
|
| 2 |
"model_name": "swinv2_base",
|
| 3 |
"model_args": {
|
| 4 |
"image_size": 448,
|
|
@@ -15,4 +16,4 @@
|
|
| 15 |
"patch_norm": true,
|
| 16 |
"layer_norm_eps": 1e-05
|
| 17 |
}
|
| 18 |
-
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"image_size": 448,
|
| 3 |
"model_name": "swinv2_base",
|
| 4 |
"model_args": {
|
| 5 |
"image_size": 448,
|
|
|
|
| 16 |
"patch_norm": true,
|
| 17 |
"layer_norm_eps": 1e-05
|
| 18 |
}
|
| 19 |
+
}
|