Skip to content

Commit

Permalink
Refactor reparameterizable layers and add GhostNetV3 (#47)
Browse files Browse the repository at this point in the history
* Merge `RepConv2D` and `MobileOneConv2D` into one `ReparameterizableConv2D`

* Add `GhostNetV3`

* Update version number

* Fix CI

* Fix array assignment issue

* Fix channels_first

* Improve test coverage
  • Loading branch information
james77777778 authored May 29, 2024
1 parent e2d31cc commit e3a9574
Show file tree
Hide file tree
Showing 24 changed files with 2,221 additions and 660 deletions.
9 changes: 9 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@

# Keras Image Models

- [Latest Updates](#latest-updates)
- [Introduction](#introduction)
- [Usage](#usage)
- [Installation](#installation)
Expand All @@ -24,6 +25,13 @@
- [License](#license)
- [Acknowledgements](#acknowledgements)

## Latest Updates

2024/05/29:

- Merge reparameterizable layers into 1 `ReparameterizableConv2D`
- Add `GhostNetV3*` from [huawei-noah/Efficient-AI-Backbones](https://github.com/huawei-noah/Efficient-AI-Backbones)

## Introduction

**K**eras **Im**age **M**odels (`kimm`) is a collection of image models, blocks and layers written in Keras 3. The goal is to offer SOTA models with pretrained weights in a user-friendly manner.
Expand Down Expand Up @@ -154,6 +162,7 @@ Reference: [Grad-CAM class activation visualization (keras.io)](https://keras.io
|EfficientNetV2|[ICML 2021](https://arxiv.org/abs/2104.00298)|`timm`|`kimm.models.EfficientNetV2*`|
|GhostNet|[CVPR 2020](https://arxiv.org/abs/1911.11907)|`timm`|`kimm.models.GhostNet*`|
|GhostNetV2|[NeurIPS 2022](https://arxiv.org/abs/2211.12905)|`timm`|`kimm.models.GhostNetV2*`|
|GhostNetV3|[arXiv 2024](https://arxiv.org/abs/2404.11202)|`github`|`kimm.models.GhostNetV3*`|
|HGNet||`timm`|`kimm.models.HGNet*`|
|HGNetV2||`timm`|`kimm.models.HGNetV2*`|
|InceptionNeXt|[arXiv 2023](https://arxiv.org/abs/2303.16900)|`timm`|`kimm.models.InceptionNeXt*`|
Expand Down
2 changes: 1 addition & 1 deletion kimm/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,4 +13,4 @@
from kimm._src.utils.model_registry import list_models
from kimm._src.version import version

__version__ = "0.2.1"
__version__ = "0.2.2"
287 changes: 0 additions & 287 deletions kimm/_src/layers/rep_conv2d.py

This file was deleted.

Loading

0 comments on commit e3a9574

Please sign in to comment.