snnn001 commited on
Commit
31008a7
·
verified ·
1 Parent(s): 92e75a2

Add LiteRT converted pvt_v2_b2

Browse files
Files changed (2) hide show
  1. README.md +52 -0
  2. model.tflite +3 -0
README.md ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: litert
3
+ base_model: timm/pvt_v2_b2.in1k
4
+ tags:
5
+ - vision
6
+ - image-classification
7
+ datasets:
8
+ - imagenet-1k
9
+ ---
10
+ # pvt_v2_b2
11
+
12
+ Converted TIMM image classification model for LiteRT.
13
+
14
+ - Source architecture: `pvt_v2_b2`
15
+ - Source checkpoint: `timm/pvt_v2_b2.in1k`
16
+ - File: `model.tflite`
17
+ - Input: `float32` tensor in NCHW layout, shape `[1, 3, 224, 224]`
18
+ - Output: ImageNet-1K logits, shape `[1, 1000]`
19
+
20
+ ## Runtime Status
21
+
22
+ - CPU smoke test: passed with LiteRT `CompiledModel`.
23
+ - GPU delegation: currently blocked for this model by rank-5 tensor patterns in the GPU backend, mostly `RESHAPE`, `TRANSPOSE`, and related window/attention operations. The model is published as CPU-ready while GPU support is being improved.
24
+
25
+ ## Model Details
26
+
27
+ - **Model Type:** Image classification / feature backbone
28
+ - **Model Stats:**
29
+ - Params (M): 25.4
30
+ - GMACs: 4.0
31
+ - Activations (M): 27.5
32
+ - Image size: 224 x 224
33
+ - **Papers:**
34
+ - PVT v2: Improved Baselines with Pyramid Vision Transformer: https://arxiv.org/abs/2106.13797
35
+ - **Dataset:** ImageNet-1k
36
+ - **Original:** https://github.com/whai362/PVT
37
+
38
+ ## Citation
39
+
40
+ ```bibtex
41
+ @article{wang2021pvtv2,
42
+ title={Pvtv2: Improved baselines with pyramid vision transformer},
43
+ author={Wang, Wenhai and Xie, Enze and Li, Xiang and Fan, Deng-Ping and Song, Kaitao and Liang, Ding and Lu, Tong and Luo, Ping and Shao, Ling},
44
+ journal={Computational Visual Media},
45
+ volume={8},
46
+ number={3},
47
+ pages={1--10},
48
+ year={2022},
49
+ publisher={Springer}
50
+ }
51
+ ```
52
+
model.tflite ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a0e6898f4d07fda3fea4d6c043f25d1cac0a2295a5de2dc0a4b93fbecc5d8a5
3
+ size 101599392