Spaces:
Sleeping
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