update nfnets

leondgarse · Sep 24, 2021 · bd4ed9a · bd4ed9a
1 parent a9a05bf
commit bd4ed9a
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/keras_cv_attention_models/nfnets/README.md b/keras_cv_attention_models/nfnets/README.md
@@ -0,0 +1,28 @@
+# ___Keras NFNets___
+***
+
+- **In progress**
+
+## Summary
+  - Keras implementation of [Github deepmind/nfnets](https://github.com/deepmind/deepmind-research/tree/master/nfnets). Paper [PDF 2102.06171 High-Performance Large-Scale Image Recognition Without Normalization](https://arxiv.org/pdf/2102.06171.pdf).
+  - Model weights reloaded from official publication.
+***
+
+## Models
+  | Model       | Params | Image  resolution | Top1 Acc | Download |
+  | ----------- | ------ | ----------------- | -------- | -------- |
+  | NFNetF0     | 71.5M  | 256               | 83.6     |          |
+  | NFNetF1     | 132.6M | 320               | 84.7     |          |
+  | NFNetF2     | 193.8M | 352               | 85.1     |          |
+  | NFNetF3     | 254.9M | 416               | 85.7     |          |
+  | NFNetF4     | 316.1M | 512               | 85.9     |          |
+  | NFNetF5     | 377.2M | 544               | 86.0     |          |
+  | NFNetF6 SAM | 438.4M | 576               | 86.5     |          |
+  | NFNetF7     | 499.5M |                   |          |          |
+## Usage
+  ```py
+  from keras_cv_attention_models import nfnets
+
+  mm = nfnets.NFNetF0()
+  mm.summary()
+  ```
diff --git a/keras_cv_attention_models/nfnets/__init__.py b/keras_cv_attention_models/nfnets/__init__.py
@@ -0,0 +1 @@
+from keras_cv_attention_models.nfnets.nfnets import NormFreeNet, NFNetF0, NFNetF1, NFNetF2, NFNetF3, NFNetF4, NFNetF5, NFNetF6, NFNetF7
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		from keras_cv_attention_models.nfnets.nfnets import NormFreeNet, NFNetF0, NFNetF1, NFNetF2, NFNetF3, NFNetF4, NFNetF5, NFNetF6, NFNetF7