transformers.js
cdcbfc12 - Add support for Segment Anything Model (#510)

Commit
1 year ago
Add support for Segment Anything Model (#510) * Update SamModel * Make `AutoModel.from_pretrained` work with SamModel * Add listed support for SAM (Segment Anything Model) * Update types of `calculateDimensions` * Throw error if reading image from tensor with dims.length != 3 * Make SamProcessor input points optional * Fix type errors * `let` -> `const` * `cat` -> `stack` * Expose `reshape_input_points` in `SamProcessor` * Add `input_labels` input parameter for SAM * Add `input_labels` to sam processor * Update SAM unit tests * Remove TODOs * Update JSDoc
References
Author
Parents
Loading