Skip to content

Commit

Permalink
Add model 2023-07-05-image_classifier_convnext_tiny_224_local_en (#13879
Browse files Browse the repository at this point in the history
)

Co-authored-by: gadde5300 <gadde5300@gmail.com>
  • Loading branch information
jsl-models and gadde5300 authored Jul 5, 2023
1 parent c999bd6 commit 27840ed
Showing 1 changed file with 75 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,75 @@
---
layout: model
title: ConvNextForImageClassification - image_classifier_convnext_tiny_224_local
author: John Snow Labs
name: image_classifier_convnext_tiny_224_local
date: 2023-07-05
tags: [convnext, en, engligh, image_classification, imagenet, convolution, open_source, tensorflow]
task: Image Classification
language: en
edition: Spark NLP 5.0.0
spark_version: 3.0
supported: true
engine: tensorflow
annotator: ConvNextForImageClassification
article_header:
type: cover
use_language_switcher: "Python-Scala-Java"
---

## Description

Pretrained ConvNext model for Image Classification, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.

The ConvNeXT model was proposed in A ConvNet for the 2020s by Zhuang Liu, Hanzi Mao, Chao-Yuan Wu, Christoph Feichtenhofer, Trevor Darrell, Saining Xie.

## Predicted Entities



{:.btn-box}
<button class="button button-orange" disabled>Live Demo</button>
<button class="button button-orange" disabled>Open in Colab</button>
[Download](https://s3.amazonaws.com/auxdata.johnsnowlabs.com/public/models/image_classifier_convnext_tiny_224_local_en_5.0.0_3.0_1688564243397.zip){:.button.button-orange.button-orange-trans.arr.button-icon}
[Copy S3 URI](s3://auxdata.johnsnowlabs.com/public/models/image_classifier_convnext_tiny_224_local_en_5.0.0_3.0_1688564243397.zip){:.button.button-orange.button-orange-trans.button-icon.button-copy-s3}

## How to use



<div class="tabs-box" markdown="1">
{% include programmingLanguageSelectScalaPythonNLU.html %}
```python
image_assembler = ImageAssembler()\
.setInputCol("image")\
.setOutputCol("image_assembler")

imageClassifier = ConvNextForImageClassification.pretrained("image_classifier_convnext_tiny_224_local", "en")\
.setInputCols("image_assembler")\
.setOutputCol("class")

pipeline = Pipeline(stages=[
image_assembler,
imageClassifier,
])

pipelineModel = pipeline.fit(imageDF)

pipelineDF = pipelineModel.transform(imageDF)
```

</div>

{:.model-param}
## Model Information

{:.table-model}
|---|---|
|Model Name:|image_classifier_convnext_tiny_224_local|
|Compatibility:|Spark NLP 5.0.0+|
|License:|Open Source|
|Edition:|Official|
|Input Labels:|[image_assembler]|
|Output Labels:|[class]|
|Language:|en|
|Size:|107.6 MB|

0 comments on commit 27840ed

Please sign in to comment.