/ Модель

Характеристики

Параметры
113 млн
Лицензия
Apache 2.0
Архитектура
vit · dense
Базовая модель
vit-base-patch16-224-in21k
Опубликована
8 ноя 2023

camera-lens-focal-length

Классификация изображенийИзображение → label

Классифицирует изображение по предполагаемому типу фокусного расстояния объектива: от ультраширокого до телеобъектива.

15скачиваний за 30 дней5лайковвеса обновлены26 фев 2024

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

Модель tonyassi/camera-lens-focal-length разработана на основе Vision Transformer ViT-Base и дообучена для классификации изображений по предполагаемому фокусному расстоянию объектива. На вход она принимает изображение и относит его к одному из пяти классов: ULTRA-WIDE, WIDE, MEDIUM, LONG-LENS или TELEPHOTO.

Модель подходит для автоматической оценки типа объектива по кадрам, в том числе для анализа изображений из кино. Для обучения использовано 5000 изображений из популярных голливудских фильмов — по 1000 на класс. Размер базовой модели составляет 113,2 млн параметров; веса доступны в формате safetensors.

Модель является дообучением google/vit-base-patch16-224-in21k и распространяется по лицензии Apache-2.0. Результат представляет собой один из пяти категориальных классов, а не точное числовое значение фокусного расстояния. Обучающие данные ограничены кадрами из популярных фильмов, поэтому качество на других типах изображений может отличаться.

Сценарии применения

  • определение типа объектива по кадру из фильма
  • классификация фотографий по предполагаемому фокусному расстоянию
  • анализ визуального стиля кинематографических кадров

Ограничения

Модель предсказывает категорию объектива, а не точное значение фокусного расстояния. Она обучена на кадрах из популярных голливудских фильмов, поэтому качество на изображениях другого происхождения может отличаться.

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

Camera Lens Focal Length

This model predicts the focal length that the camera lens used to capture an image. It takes in an image and returns one of the following labels:

  • ULTRA-WIDE
  • WIDE
  • MEDIUM
  • LONG-LENS
  • TELEPHOTO

How to use

from transformers import pipeline

pipe = pipeline("image-classification", model="tonyassi/camera-lens-focal-length")
result = pipe('image.png')

print(result)

Dataset

Trained on a total of 5000 images. 1000 images from each label. Images were taken from popular Hollywood movies.

ULTRA-WIDE

image/jpeg

WIDE

image/jpeg

MEDIUM

image/jpeg

LONG-LENS

image/jpeg

TELEPHOTO

image/jpeg

Model description

This model is a fine-tuned version of google/vit-base-patch16-224-in21k.

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 8
  • eval_batch_size: 8
  • seed: 42
  • gradient_accumulation_steps: 2
  • total_train_batch_size: 16
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • lr_scheduler_warmup_ratio: 0.1
  • num_epochs: 5

Framework versions

  • Transformers 4.35.0
  • Pytorch 2.1.0+cu118
  • Datasets 2.14.6
  • Tokenizers 0.14.1

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · image-classification

Можно запустить

Подбираем GPU…

Класс архитектуры
ViTForImageClassification
Ревизия весов
4df9475

0 моделей