Model Overview

Description:

Hive’s AI-Generated Image Classification model is optimized for use with the kind of media generated by popular AI generative engines such as DALL-E, Midjourney, and Stable Diffusion.

The model has two heads. The first head gives a binary classification for all images, identifying whether or not they were AI generated and the accompanying confidence score. The second head provides further details as to the image's source, with support for the most popular AI art generators currently in use. If the model cannot identify a source, it will return none under the source head.

This model is ready for commercial and non-commercial use.

Third-Party Community Consideration

This model is not owned or developed by NVIDIA. This model has been developed and built to a third-party’s requirements for this application and use case.

License/Terms of Use:

https://thehive.ai/terms-of-use

Model Architecture:

Architecture Type: Convolution Neural Network (CNN)]

Network Architecture: EfficientNet-B4

Input:

Input Type(s): Ensemble Gradient Boosted (EGB) Image

Input Format(s): PNG, JPEG, or JPG

Input Parameters: Two-Dimensional (2D)

Other Properties Related to Input: [Specific Resolution/Minimum or Maximum Resolution, Characters (Including Restrictions), or Tokens; Image Range Needed (W x Y x Z), Pre-Processing Needed, Alpha Channel, Bit; If None Applicable, Please State Such]

Output:

Output Type(s): Text

Output Format: String

Output Parameters: One Dimensional (1D)

Software Integration:

Supported Hardware Microarchitecture Compatibility:

NVIDIA Ampere
NVIDIA Jetson
NVIDIA Hopper
NVIDIA Lovelace
NVIDIA Volta

[Preferred/Supported] Operating System(s): Linux

Model Version(s):

v1.0

Training Dataset:

** Data Collection Method by dataset

Unknown

** Labeling Method by dataset

Unknown

Properties (Quantity, Dataset Descriptions, Sensor(s)): Trained on a large dataset comprising millions of artificially generated images and human-created images such as photographs, digital and traditional art, and memes sourced from across the web.

Inference:

Engine: TensorRT

Test Hardware: L40

Ethical Considerations (For NVIDIA Models Only):

NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. When downloaded or used in accordance with our terms of service, developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse.

Please report security vulnerabilities or NVIDIA AI Concerns here.