Commit ·
a9e2ffc
1
Parent(s): c0931f3
add comparisons, fix typos
Browse files
README.md
CHANGED
|
@@ -6,7 +6,8 @@ tags:
|
|
| 6 |
license: cc-by-nc-4.0
|
| 7 |
---
|
| 8 |
|
| 9 |
-
To be clear, this model is tailored to my image and video classification tasks, not to imagenet.
|
|
|
|
| 10 |
|
| 11 |
## Model Details
|
| 12 |
- **Model tasks:** Image classification / video classification / feature backbone
|
|
@@ -55,4 +56,9 @@ traced_model.save(model_name)
|
|
| 55 |
|
| 56 |
# Load the training-ready model
|
| 57 |
model = torch.load(model_name)
|
| 58 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
license: cc-by-nc-4.0
|
| 7 |
---
|
| 8 |
|
| 9 |
+
To be clear, this model is tailored to my image and video classification tasks, not to imagenet.
|
| 10 |
+
I built EfficientNetV2.5 to outperform the existing EfficientNet b0 to b4 and EfficientNetV2 t to l models, whether in TensorFlow or PyTorch, in terms of top-1 accuracy, efficiency, and robustness on my datasets and GVNS benchmarks.
|
| 11 |
|
| 12 |
## Model Details
|
| 13 |
- **Model tasks:** Image classification / video classification / feature backbone
|
|
|
|
| 56 |
|
| 57 |
# Load the training-ready model
|
| 58 |
model = torch.load(model_name)
|
| 59 |
+
```
|
| 60 |
+
|
| 61 |
+
# Top-1 Accuracy Comparisons
|
| 62 |
+
I will publish the results in another model repository, including the link to the GVNS benchmark.
|
| 63 |
+
|
| 64 |
+
Note that `efficientnet_b3_pruned` achieved the second highest top-1 accuracy as well as the highest epoch-1 training accuracy on my task, out of all previous EfficientNet models my 24 GB VRAM RTX 3090 could handle.
|