Пошук аналогів за фото: AI-розпізнавання меблів у мобільному додатку
Клієнт хоче, щоб користувач сфотографував диван та отримав схожі моделі з цінами. Але меблі — не одяг: однакові на вигляд крісла можуть відрізнятися за стилем і розмірами. Помилка на етапі класифікації зображень веде до хибних рекомендацій. Ми вирішуємо це завдання за допомогою комбінації Transfer Learning на згорткових мережах та векторного пошуку. Наш досвід показує, що правильно налаштована модель економить до 40% часу на ручній обробці каталогу, а вартість підтримки каталогу знижується на 25% за рахунок автоматизації.
Чому меблі складніші за одяг?
У меблів жорстка геометрія та впізнавані форми — це спрощує класифікацію. Але пошук «схожого за менші гроші» вимагає не лише візуального збігу, але й розуміння стилю (скандинавський мінімалізм, лофт, класика) та масштабу. Диван на фото без контексту не дозволяє зрозуміти, тримісний він чи двомісний. Ми вирішуємо це за допомогою додаткових класифікаторів стилю меблів та фільтрації за атрибутами.
Джерело: рекомендації з обробки зображень меблів з відкритих датасетів (наприклад, Furniture-180).
Проблеми, які ми вирішуємо
- Категоризація: Більшість готових моделей (Google Cloud Vision, AWS Rekognition) навчені на ImageNet — там меблів достатньо. Але точність падає для рідкісних форм. Ми донавчаємо EfficientNet або MobileNetV3 на каталозі магазину: 50 000 зображень (по 200–300 на категорію) дають впевнену класифікацію основних категорій: диван, крісло, стіл, стілець, шафа, ліжко, тумба.
- Визначення стилю: Категорія — лише перший крок. Для пошуку аналогів важливий стиль. Використовуємо CLIP, який розуміє текстові описи: "скандинавський мінімалізм", "лофт", "класика". CLIP порівнює ембеддінг зображення з ембеддінгами текстових стилів і повертає найбільш імовірний.
- Пошук з фільтрами: Архітектура та ж, що й для одягу: ембеддінг → vector search. Але для меблів важлива фільтрація за матеріалом (дерево, метал, тканина), кольором, розмірною групою. Розмірний клас без AR визначається приблизно — за співвідношенням сторін і типовими пропорціями для категорії.
Порівняння підходів до класифікації
| Підхід | Точність | Час на навчання | Набір даних |
|---|---|---|---|
| Готовий API (Google Cloud Vision) | 75-85% | 0 | Не потрібно |
| Transfer Learning (MobileNetV3) | 92-96% | 2-4 дні | 500–1000 на категорію |
| Навчання з нуля (ResNet) | 96%+ | 2-4 тижні | 10000+ на категорію |
Transfer Learning — оптимальний вибір для більшості рітейлерів: точність 92% при мінімальних витратах часу. Він точніший за готовий API в середньому на 15% за нашими тестами. При цьому Transfer Learning потребує в 2-3 рази менше даних, ніж навчання з нуля, при порівнянній точності. Ми використовуємо MobileNetV3 або EfficientNet-Lite — вони працюють безпосередньо на пристрої.
Як визначити стиль меблів?
| Метод | Точність визначення стилю | Необхідність розмітки |
|---|---|---|
| Тільки візуальні ознаки (CNN) | 70-80% | 500+ зображень на стиль |
| CLIP (текст+зображення) | 85-95% | Текстові описи стилів |
| Гібрид (CLIP + кастомний класифікатор) | 90-97% | 200 зображень на стиль |
CLIP дозволяє швидко підлаштуватися під новий стиль без перенавчання — достатньо додати текстовий опис. Докладніше про CLIP можна дізнатися в офіційному репозиторії.
Як ми це робимо: стек та приклади
Для Android використовуємо TFLite з XNNPACK оптимізацією. Приклад класифікатора:
// Android: TFLite классификация мебели
class FurnitureClassifier(context: Context) {
private val interpreter: Interpreter by lazy {
val model = FileUtil.loadMappedFile(context, "furniture_classifier_v2.tflite")
Interpreter(model, Interpreter.Options().apply {
numThreads = 4
useXNNPACK = true
})
}
fun classify(bitmap: Bitmap): List<FurnitureClassification> {
val resized = Bitmap.createScaledBitmap(bitmap, 224, 224, true)
val input = TensorImage.fromBitmap(resized)
val output = TensorBuffer.createFixedSize(intArrayOf(1, NUM_CLASSES), DataType.FLOAT32)
interpreter.run(input.buffer, output.buffer)
return output.floatArray
.mapIndexed { index, score -> FurnitureClassification(LABELS[index], score) }
.filter { it.score > 0.1f }
.sortedByDescending { it.score }
}
}
Для iOS — Core ML. CLIP дозволяє визначати стиль через текстові промпти:
// iOS: CLIP-based style detection через Core ML
// CLIP модель конвертирована в .mlpackage
func detectStyle(_ image: UIImage) async throws -> [StyleScore] {
let styleDescriptions = [
"scandinavian minimalist furniture",
"industrial loft style furniture",
"classic traditional furniture",
"mid-century modern furniture",
"boho eclectic furniture"
]
// CLIP сравнивает image embedding с text embeddings стилей
let imageEmbedding = try await clipEncoder.encodeImage(image)
return styleDescriptions.enumerated().map { i, desc in
let textEmbedding = clipEncoder.encodeText(desc)
let similarity = cosineSimilarity(imageEmbedding, textEmbedding)
return StyleScore(style: desc, score: similarity)
}.sorted { $0.score > $1.score }
}
Як працює пошук аналогів по каталогу?
Ембеддінги товарів зберігаються у векторній БД (наприклад, FAISS або Pinecone). Пошук включає кілька етапів:
- Вилучення ембеддінгу: зображення користувача проходить через ту ж модель, що й каталог.
- Векторний пошук: косинусна відстань до всіх ембеддінгів каталогу.
- Застосування фільтрів: відсів за категорією, стилем, матеріалом, кольором, розміром та ціною.
- Постобробка: сортування за схожістю та виведення топ-10 результатів.
struct FurnitureSearchFilters {
let category: FurnitureCategory
let style: StyleTag?
let colorFamily: ColorFamily? // warm, cool, neutral
let material: MaterialType? // wood, metal, upholstered
let maxDimensionClass: SizeClass? // compact, standard, large
let priceRange: ClosedRange<Int>?
let inStockOnly: Bool
}
Процес інтеграції
Впровадження функції розпізнавання та пошуку аналогів проходить кілька етапів:
Докладніше про етапи
- Аналіз каталогу: оцінка обсягу зображень, категорій, стилів.
- Підготовка даних: розмітка 200–500 зображень на категорію/стиль (якщо потрібне донавчання).
- Навчання моделі: Transfer Learning на TensorFlow/Keras або Create ML.
- Інтеграція в додаток: підключення TFLite/Core ML, налаштування векторного сховища.
- Тестування: A/B тест точності розпізнавання та швидкості пошуку.
- Деплой: реліз через App Store/Google Play, моніторинг.
Мінімально життєздатний продукт (MVP) можна запустити за 1 тиждень, використовуючи готове API розпізнавання та хмарне векторне сховище. Повне рішення з власною моделлю, CLIP та AR-розмірами займає 1–2 місяці. Наприклад, для одного з клієнтів з каталогом 10 000 товарів ми донавчили MobileNetV3 на 500 зображеннях на категорію. Точність класифікації зросла з 72% до 94%.
Обсяг робіт
- MVP: інтеграція з готовим API (Google Cloud Vision + векторний пошук) — термін від 1 тижня.
- Повне рішення: донавчена TFLite/CoreML модель, CLIP-класифікатор стилю, векторне сховище з фільтрами, AR-розміри (LiDAR) — термін 1–2 місяці.
- Документація по моделі та API.
- Навчання команди замовника.
- Пост-релізна підтримка 2 тижні.
Наша команда має 7+ років досвіду в мобільній розробці та машинному навчанні. Ми виконали 30+ проектів з розпізнавання зображень. Гарантуємо якість: модель працюватиме офлайн, без затримок. Якщо ви хочете реалізувати аналогічну функцію, зв'яжіться з нами — ми підберемо оптимальне рішення. Отримайте консультацію з інтеграції — оцінимо ваш проект безкоштовно.







