IMvision12 commited on
Commit
e84bd7c
·
verified ·
1 Parent(s): fbf9dc1

fix readme.md

Browse files
Files changed (1) hide show
  1. README.md +57 -8
README.md CHANGED
@@ -8,20 +8,69 @@ tags:
8
  - kerasformers
9
  - image-classification
10
  - inceptionv4
11
- - tf
12
- - jax
13
  - pytorch
 
 
14
  ---
15
 
16
- # inception_v4_tf_in1k
 
 
 
 
 
 
 
 
 
 
 
 
17
 
18
- Pure-Keras 3 image-classification weight for [kerasformers](https://github.com/IMvision12/KerasFormers), converted from [timm/inception_v4.tf_in1k](https://huggingface.co/timm/inception_v4.tf_in1k).
19
 
20
- ## Usage
 
 
21
 
22
  ```python
23
- from kerasformers.models.inceptionv4 import InceptionV4ImageClassify
24
- model = InceptionV4ImageClassify.from_weights("inception_v4_tf_in1k")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
  ```
26
 
27
- License: **apache-2.0**, inherited from the upstream source.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  - kerasformers
9
  - image-classification
10
  - inceptionv4
11
+ - backbone
12
+ - arxiv:1602.07261
13
  - pytorch
14
+ - jax
15
+ - tf
16
  ---
17
 
18
+ ## ***See [our collection](https://huggingface.co/collections/kerasformers/inception-v4-6a6bd726e3166c8a5afe840e) for all versions of Inception-V4.***
19
+
20
+ # Run Inception-V4 with Keras 3: JAX, PyTorch, or TensorFlow
21
+
22
+ [![GitHub](https://img.shields.io/badge/GitHub-KerasFormers-black?logo=github)](https://github.com/IMvision12/KerasFormers) [![Docs](https://img.shields.io/badge/Docs-Inception--V4-blue)](https://imvision12.github.io/KerasFormers/classification_backbones/) [![Collection](https://img.shields.io/badge/HF-Inception--V4%20collection-yellow)](https://huggingface.co/collections/kerasformers/inception-v4-6a6bd726e3166c8a5afe840e)
23
+
24
+ # kerasformers/inception_v4_tf_in1k
25
+
26
+ Paper: [Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning (arXiv:1602.07261)](https://arxiv.org/abs/1602.07261) · [HF Papers](https://huggingface.co/papers/1602.07261)
27
+
28
+ Inception-V4 is a uniform Inception architecture without residual connections. Classifier or backbone.
29
+
30
+ For more details on the model, please go to the upstream [model card](https://huggingface.co/timm/inception_v4.tf_in1k).
31
 
32
+ Pure-**Keras 3** conversion of [`timm/inception_v4.tf_in1k`](https://huggingface.co/timm/inception_v4.tf_in1k) for [kerasformers](https://github.com/IMvision12/KerasFormers). One implementation runs unmodified on **TensorFlow / Torch / JAX**.
33
 
34
+ This is an **image-classification / backbone** checkpoint (`InceptionV4ImageClassify` / `InceptionV4Model`).
35
+
36
+ ## ✨ Quick start
37
 
38
  ```python
39
+ import os
40
+ os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
41
+
42
+ from PIL import Image
43
+ import numpy as np
44
+ from kerasformers.models.inceptionv4 import InceptionV4ImageClassify, InceptionV4Model
45
+
46
+ model = InceptionV4ImageClassify.from_weights("kerasformers/inception_v4_tf_in1k")
47
+ backbone = InceptionV4Model.from_weights(
48
+ "kerasformers/inception_v4_tf_in1k", as_backbone=True
49
+ )
50
+
51
+ image = Image.open("your_image.jpg").convert("RGB")
52
+ image = image.resize((299, 299))
53
+ x = np.asarray(image, dtype="float32")[None] # (1, H, W, 3)
54
+ print(model(x).shape) # (1, num_classes)
55
+ feats = backbone(x)
56
+ print(len(feats), [tuple(f.shape) for f in feats])
57
  ```
58
 
59
+ Load any Inception-V4 variant the same way with `from_weights("kerasformers/<variant>")`:
60
+
61
+ | Variant | Hub |
62
+ |---|---|
63
+ | `inception_v4_tf_in1k` | [`kerasformers/inception_v4_tf_in1k`](https://huggingface.co/kerasformers/inception_v4_tf_in1k) |
64
+
65
+ ## Tips
66
+
67
+ - Set `KERAS_BACKEND` **before** importing Keras / kerasformers.
68
+ - `InceptionV4ImageClassify` returns class logits; `InceptionV4Model` returns features (`as_backbone=True` for multi-scale stages).
69
+ - See [docs](https://imvision12.github.io/KerasFormers/classification_backbones/) and [Loading Weights](https://imvision12.github.io/KerasFormers/loading_weights/).
70
+ - Upstream / timm checkpoints: `InceptionV4ImageClassify.from_weights("hf:timm/inception_v4.tf_in1k")`.
71
+
72
+ ## Special Thanks
73
+
74
+ A huge thank you to the Inception-V4 authors and the timm / Hub communities for creating and releasing these models.
75
+
76
+ License: see YAML `license` (usually matches the upstream checkpoint).