---
title: Amazon Products Demo
emoji: 📈
colorFrom: indigo
colorTo: pink
sdk: gradio
sdk_version: 3.41.2
app_file: app.py
pinned: false
---
Description:

We present a demo for performing object segmentation using a model trained on Amazon's ARMBench dataset. The model was trained on over 37,000 training images and validated on 4,425 images.

Usage:

You can use our demo by uploading your product image, and it will provide you with a segmented image.

Dataset:

-The model was trained on the ARMBench segmentation dataset, which comprises more than 50,000 images. 

-License: Creative Commons

-Paper: ARMBench: An object-centric benchmark dataset for robotic manipulation

-Authors: Chaitanya Mitash, Fan Wang, Shiyang Lu, Vikedo Terhuja, Tyler Garaas, Felipe Polido, Manikantan Nambi


You can learn more about this dataset on https://www.amazon.science/blog/amazon-releases-largest-dataset-for-training-pick-and-place-robots.


Download Dataset:

To download the dataset we used, you can use the following command in colab :

!wget https://armbench-dataset.s3.amazonaws.com/segmentation/armbench-segmentation-0.1.tar.gz


Feel free to explore and use this repository for your object segmentation needs. If you have any questions or need assistance, please don't hesitate to reach out