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)

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:

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.