From 96129b03e7e0f06e50c2595fad3adc5a33843dc4 Mon Sep 17 00:00:00 2001 From: awsaf49 Date: Wed, 27 Jul 2022 15:06:26 +0600 Subject: [PATCH] Prepare for `v1.0.7` --- README.md | 13 +++++++++++-- gcvit/models/gcvit.py | 2 +- gcvit/version.py | 2 +- 3 files changed, 13 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 9f6bc5a..39ce5db 100644 --- a/README.md +++ b/README.md @@ -32,10 +32,18 @@ This library implements GCViT using Tensorflow 2.0 specifally in tf ## Result -> The reported result in the paper is shown in the figure. But due to issues in the **codebase** actual result differs from the reported result. - +Official codebase had some issue which has been fixed recently (27 July 2022). Here's the result of ported weights on **ImageNetV2-Test** data, + +| Model | Acc@1 | Acc@5 | #Params | +|--------------|-------|-------|---------| +| GCViT-XXTiny | 63 | 85 | 12M | +| GCViT-XTiny | 66 | 87 | 20M | +| GCViT-Tiny | 69 | 89 | 28M | +| GCViT-Small | 69 | 89 | 51M | +| GCViT-Base | 71 | 90 | 90M | + ## Installation ```bash pip install -U gcvit @@ -102,6 +110,7 @@ Here is grad-cam result after training on Flower Classification Dataset, ## To Do +- [x] New updated weights have been added. - [x] Working training example in Colab & Kaggle. - [x] GradCAM showcase. - [x] Gradio Demo. diff --git a/gcvit/models/gcvit.py b/gcvit/models/gcvit.py index 3b04c4a..8982894 100644 --- a/gcvit/models/gcvit.py +++ b/gcvit/models/gcvit.py @@ -5,7 +5,7 @@ BASE_URL = 'https://github.com/awsaf49/gcvit-tf/releases/download' -TAG = 'v1.0.4' +TAG = 'v1.0.7' NAME2CONFIG = { 'gcvit_xxtiny': {'window_size': (7, 7, 14, 7), 'dim': 64, diff --git a/gcvit/version.py b/gcvit/version.py index 010279a..84ce141 100644 --- a/gcvit/version.py +++ b/gcvit/version.py @@ -1 +1 @@ -__version__ = "1.0.6" \ No newline at end of file +__version__ = "1.0.7" \ No newline at end of file