ThereforeGames
8be73292d6
Major refactor: rewrite MaskArbiter to accept various tensor/list mask shapes, use efficient single-channel evaluation, deterministic resizing for sorting, improved sorting (leftmost/topmost/innermost/largest/random/merged), reverse support, and proper empty-batch behavior (returns black selected mask and preserves ordered masks). GroundingDINO/SAM2 node: switch to SAM2ImagePredictor, continue on images with no detections, fix box scaling and mask creation, reduce copies, and return correctly shaped outputs. Also: bump package version, update README/CHANGELOG, and add unit tests for MaskArbiter behavior.
12 lines
346 B
Python
12 lines
346 B
Python
from .mask_arbiter import *
|
|
from .groundingdinosam2segmentlist import *
|
|
|
|
__version__ = "0.3.0"
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"MaskArbiter": MaskArbiter,
|
|
"GroundingDinoSAM2SegmentList": GroundingDinoSAM2SegmentList,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {"MaskArbiter": "Mask Arbiter", "GroundingDinoSAM2SegmentList": "GroundingDinoSAM2SegmentList"}
|