This is the code to implement Segment Anything (SAM) using TensorRT(C++).
We use MobileSAM as SAM in order to be suitable for lightweight deployment of applications.
/usr/local/TensorRT-8.5.3.1/bin/trtexec -- --saveEngine=./models/mobile_sam_encoder.engine --memPoolSize=workspace:10240
/usr/local/TensorRT-8.5.3.1/bin/trtexec -- --minShapes=point_coords:1x1x2,point_labels:1x1 --optShapes=point_coords:1x1x2,point_labels:1x1 --maxShapes=point_coords:1x10x2,point_labels:1x10 --saveEngine=./models/mobile_sam_decoder.engine
mkdir build
cd build
cmake ..
make -j8
./sam