Skip to content

Real time inference using FastSAM for bin picking computer vision task

Notifications You must be signed in to change notification settings

mora-bprs/bin-picking-real-time

Repository files navigation

Bin-Picking-Real-Time

Here's the real-time video demonstration using the Fast-SAM model.

Watch the video on YouTube

Getting Started

There are two ways that you can use the model:

  1. Using Google Colab.
  • You can have a look at the notebook and run it in Google Colab.
  • Not recommended for real-time inference.
  1. Using a local environment for real-time inference.
  • You can clone the repository and run the python script for real-time inference.

1. Google Colab

  • Open the notebook in Colab

Open in Google Colab

  • Modify the COLAB and INITIALIZED variables accordingly.
  • Menu -> Runtime -> Run All -> Sit back and relax.

2. Local Environment for Real-time Inference (For Windows)

  • Clone the repository
git clone https://github.com/mora-bprs/bin-picking-real-time.git
cd bin-picking-real-time
  • Create a python environment
python -m venv bin-venv
bin-venv\Scripts\Activate.ps1
  • Install the required packages. About 300MB of data will be downloaded.
pip install -r requirements.txt
  • Run the python script for real-time inference
python smooth_main_rt.py
  • Press 'q' to exit the real-time inference.

  • Run the following command to deactivate the virtual environment

deactivate

Possible Errors & Solutions

  • You have to install python and venv if not installed
  • Python version used: 3.10.7
  • Install all the requirements in a virtual environment and then run the scripts
  • Change camera index if there are multiple cameras connected to the system (default is 0)

Model Checkpoints

Source: https://pypi.org/project/segment-anything-fast/

Click the links below to download the checkpoint for the corresponding model type.

About

Real time inference using FastSAM for bin picking computer vision task

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages