FredZhang7
commited on
Commit
•
19fce4d
1
Parent(s):
7db28a3
add model details
Browse files
README.md
CHANGED
@@ -8,6 +8,18 @@ license: cc-by-nc-4.0
|
|
8 |
|
9 |
To be clear, this model is tailored to my image and video classification tasks, not to imagenet. I built EfficientNetV2.5 to outperform the existing EfficientNet b0 to b7 and EfficientNetV2 t to xl models, whether in TensorFlow or PyTorch, in terms of top-1 accuracy, efficiency, and robustness on my datasets and benchmarks.
|
10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
|
12 |
To change the number of classes, replace the linear classification layer.
|
13 |
Here's an example to convert the architecture into a training-ready model.
|
|
|
8 |
|
9 |
To be clear, this model is tailored to my image and video classification tasks, not to imagenet. I built EfficientNetV2.5 to outperform the existing EfficientNet b0 to b7 and EfficientNetV2 t to xl models, whether in TensorFlow or PyTorch, in terms of top-1 accuracy, efficiency, and robustness on my datasets and benchmarks.
|
10 |
|
11 |
+
## Model Details
|
12 |
+
- **Model tasks:** Image classification / video classification / feature backbone
|
13 |
+
- **Model stats:**
|
14 |
+
- Params: 16.64 M
|
15 |
+
- Multiply-Add Operations: 5.32 G
|
16 |
+
- Image size: train = 299x299 / 304x304, test = 304x304
|
17 |
+
- **Papers:**
|
18 |
+
- EfficientNetV2: Smaller Models and Faster Training: https://arxiv.org/abs/2104.00298
|
19 |
+
- Layer-adaptive sparsity for the Magnitude-based Pruning: https://arxiv.org/abs/2010.07611
|
20 |
+
- **Dataset:** ImageNet-1k
|
21 |
+
- **Pretrained:** Yes, but requires finetuning
|
22 |
+
- **Original:** This model architecture is original
|
23 |
|
24 |
To change the number of classes, replace the linear classification layer.
|
25 |
Here's an example to convert the architecture into a training-ready model.
|