Spaces:

hugging-apps
/

aq3d-instance-segmentation

Running on Zero

App Files Files Community

aq3d-instance-segmentation / labels.py

multimodalart HF Staff

Upload folder using huggingface_hub

9550667 verified about 9 hours ago

Raw

History Blame Contribute Delete

14.8 kB

	"""ScanNet200 instance-segmentation classes.

	The 200 ScanNet200 categories minus the two non-instance ones (``wall``,
	``floor``), in ascending ScanNet class-id order -- exactly the ordering AQ3D's
	``label_database.yaml`` produces. Colors are the official
	``SCANNET_COLOR_MAP_200`` entries.
	"""

	CLASS_NAMES = [
	"chair",
	"table",
	"door",
	"couch",
	"cabinet",
	"shelf",
	"desk",
	"office chair",
	"bed",
	"pillow",
	"sink",
	"picture",
	"window",
	"toilet",
	"bookshelf",
	"monitor",
	"curtain",
	"book",
	"armchair",
	"coffee table",
	"box",
	"refrigerator",
	"lamp",
	"kitchen cabinet",
	"towel",
	"clothes",
	"tv",
	"nightstand",
	"counter",
	"dresser",
	"stool",
	"cushion",
	"plant",
	"ceiling",
	"bathtub",
	"end table",
	"dining table",
	"keyboard",
	"bag",
	"backpack",
	"toilet paper",
	"printer",
	"tv stand",
	"whiteboard",
	"blanket",
	"shower curtain",
	"trash can",
	"closet",
	"stairs",
	"microwave",
	"stove",
	"shoe",
	"computer tower",
	"bottle",
	"bin",
	"ottoman",
	"bench",
	"board",
	"washing machine",
	"mirror",
	"copier",
	"basket",
	"sofa chair",
	"file cabinet",
	"fan",
	"laptop",
	"shower",
	"paper",
	"person",
	"paper towel dispenser",
	"oven",
	"blinds",
	"rack",
	"plate",
	"blackboard",
	"piano",
	"suitcase",
	"rail",
	"radiator",
	"recycling bin",
	"container",
	"wardrobe",
	"soap dispenser",
	"telephone",
	"bucket",
	"clock",
	"stand",
	"light",
	"laundry basket",
	"pipe",
	"clothes dryer",
	"guitar",
	"toilet paper holder",
	"seat",
	"speaker",
	"column",
	"bicycle",
	"ladder",
	"bathroom stall",
	"shower wall",
	"cup",
	"jacket",
	"storage bin",
	"coffee maker",
	"dishwasher",
	"paper towel roll",
	"machine",
	"mat",
	"windowsill",
	"bar",
	"toaster",
	"bulletin board",
	"ironing board",
	"fireplace",
	"soap dish",
	"kitchen counter",
	"doorframe",
	"toilet paper dispenser",
	"mini fridge",
	"fire extinguisher",
	"ball",
	"hat",
	"shower curtain rod",
	"water cooler",
	"paper cutter",
	"tray",
	"shower door",
	"pillar",
	"ledge",
	"toaster oven",
	"mouse",
	"toilet seat cover dispenser",
	"furniture",
	"cart",
	"storage container",
	"scale",
	"tissue box",
	"light switch",
	"crate",
	"power outlet",
	"decoration",
	"sign",
	"projector",
	"closet door",
	"vacuum cleaner",
	"candle",
	"plunger",
	"stuffed animal",
	"headphones",
	"dish rack",
	"broom",
	"guitar case",
	"range hood",
	"dustpan",
	"hair dryer",
	"water bottle",
	"handicap bar",
	"purse",
	"vent",
	"shower floor",
	"water pitcher",
	"mailbox",
	"bowl",
	"paper bag",
	"alarm clock",
	"music stand",
	"projector screen",
	"divider",
	"laundry detergent",
	"bathroom counter",
	"object",
	"bathroom vanity",
	"closet wall",
	"laundry hamper",
	"bathroom stall door",
	"ceiling light",
	"trash bin",
	"dumbbell",
	"stair rail",
	"tube",
	"bathroom cabinet",
	"cd case",
	"closet rod",
	"coffee kettle",
	"structure",
	"shower head",
	"keyboard piano",
	"case of water bottles",
	"coat rack",
	"storage organizer",
	"folded chair",
	"fire alarm",
	"power strip",
	"calendar",
	"poster",
	"potted plant",
	"luggage",
	"mattress"
	]

	CLASS_COLORS = [
	[
	188.0,
	189.0,
	34.0
	],
	[
	255.0,
	152.0,
	150.0
	],
	[
	214.0,
	39.0,
	40.0
	],
	[
	91.0,
	135.0,
	229.0
	],
	[
	31.0,
	119.0,
	180.0
	],
	[
	229.0,
	91.0,
	104.0
	],
	[
	247.0,
	182.0,
	210.0
	],
	[
	91.0,
	229.0,
	110.0
	],
	[
	255.0,
	187.0,
	120.0
	],
	[
	141.0,
	91.0,
	229.0
	],
	[
	112.0,
	128.0,
	144.0
	],
	[
	196.0,
	156.0,
	148.0
	],
	[
	197.0,
	176.0,
	213.0
	],
	[
	44.0,
	160.0,
	44.0
	],
	[
	148.0,
	103.0,
	189.0
	],
	[
	229.0,
	91.0,
	223.0
	],
	[
	219.0,
	219.0,
	141.0
	],
	[
	192.0,
	229.0,
	91.0
	],
	[
	88.0,
	218.0,
	137.0
	],
	[
	58.0,
	98.0,
	137.0
	],
	[
	177.0,
	82.0,
	239.0
	],
	[
	255.0,
	127.0,
	14.0
	],
	[
	237.0,
	204.0,
	37.0
	],
	[
	41.0,
	206.0,
	32.0
	],
	[
	62.0,
	143.0,
	148.0
	],
	[
	34.0,
	14.0,
	130.0
	],
	[
	143.0,
	45.0,
	115.0
	],
	[
	137.0,
	63.0,
	14.0
	],
	[
	23.0,
	190.0,
	207.0
	],
	[
	16.0,
	212.0,
	139.0
	],
	[
	90.0,
	119.0,
	201.0
	],
	[
	125.0,
	30.0,
	141.0
	],
	[
	150.0,
	53.0,
	56.0
	],
	[
	186.0,
	197.0,
	62.0
	],
	[
	227.0,
	119.0,
	194.0
	],
	[
	38.0,
	100.0,
	128.0
	],
	[
	120.0,
	31.0,
	243.0
	],
	[
	154.0,
	59.0,
	103.0
	],
	[
	169.0,
	137.0,
	78.0
	],
	[
	143.0,
	245.0,
	111.0
	],
	[
	37.0,
	230.0,
	205.0
	],
	[
	14.0,
	16.0,
	155.0
	],
	[
	196.0,
	51.0,
	182.0
	],
	[
	237.0,
	80.0,
	38.0
	],
	[
	138.0,
	175.0,
	62.0
	],
	[
	158.0,
	218.0,
	229.0
	],
	[
	38.0,
	96.0,
	167.0
	],
	[
	190.0,
	77.0,
	246.0
	],
	[
	208.0,
	49.0,
	84.0
	],
	[
	208.0,
	193.0,
	72.0
	],
	[
	55.0,
	220.0,
	57.0
	],
	[
	10.0,
	125.0,
	140.0
	],
	[
	76.0,
	38.0,
	202.0
	],
	[
	191.0,
	28.0,
	135.0
	],
	[
	211.0,
	120.0,
	42.0
	],
	[
	118.0,
	174.0,
	76.0
	],
	[
	17.0,
	242.0,
	171.0
	],
	[
	20.0,
	65.0,
	247.0
	],
	[
	208.0,
	61.0,
	222.0
	],
	[
	162.0,
	62.0,
	60.0
	],
	[
	210.0,
	235.0,
	62.0
	],
	[
	45.0,
	152.0,
	72.0
	],
	[
	35.0,
	107.0,
	149.0
	],
	[
	160.0,
	89.0,
	237.0
	],
	[
	227.0,
	56.0,
	125.0
	],
	[
	169.0,
	143.0,
	81.0
	],
	[
	42.0,
	143.0,
	20.0
	],
	[
	25.0,
	160.0,
	151.0
	],
	[
	82.0,
	75.0,
	227.0
	],
	[
	253.0,
	59.0,
	222.0
	],
	[
	240.0,
	130.0,
	89.0
	],
	[
	123.0,
	172.0,
	47.0
	],
	[
	71.0,
	194.0,
	133.0
	],
	[
	24.0,
	94.0,
	205.0
	],
	[
	134.0,
	16.0,
	179.0
	],
	[
	159.0,
	32.0,
	52.0
	],
	[
	213.0,
	208.0,
	88.0
	],
	[
	64.0,
	158.0,
	70.0
	],
	[
	18.0,
	163.0,
	194.0
	],
	[
	65.0,
	29.0,
	153.0
	],
	[
	177.0,
	10.0,
	109.0
	],
	[
	152.0,
	83.0,
	7.0
	],
	[
	83.0,
	175.0,
	30.0
	],
	[
	18.0,
	199.0,
	153.0
	],
	[
	61.0,
	81.0,
	208.0
	],
	[
	213.0,
	85.0,
	216.0
	],
	[
	170.0,
	53.0,
	42.0
	],
	[
	161.0,
	192.0,
	38.0
	],
	[
	23.0,
	241.0,
	91.0
	],
	[
	12.0,
	103.0,
	170.0
	],
	[
	151.0,
	41.0,
	245.0
	],
	[
	133.0,
	51.0,
	80.0
	],
	[
	184.0,
	162.0,
	91.0
	],
	[
	50.0,
	138.0,
	38.0
	],
	[
	31.0,
	237.0,
	236.0
	],
	[
	39.0,
	19.0,
	208.0
	],
	[
	223.0,
	27.0,
	180.0
	],
	[
	254.0,
	141.0,
	85.0
	],
	[
	97.0,
	144.0,
	39.0
	],
	[
	106.0,
	231.0,
	176.0
	],
	[
	12.0,
	61.0,
	162.0
	],
	[
	124.0,
	66.0,
	140.0
	],
	[
	137.0,
	66.0,
	73.0
	],
	[
	250.0,
	253.0,
	26.0
	],
	[
	55.0,
	191.0,
	73.0
	],
	[
	60.0,
	126.0,
	146.0
	],
	[
	153.0,
	108.0,
	234.0
	],
	[
	184.0,
	58.0,
	125.0
	],
	[
	135.0,
	84.0,
	14.0
	],
	[
	139.0,
	248.0,
	91.0
	],
	[
	53.0,
	200.0,
	172.0
	],
	[
	63.0,
	69.0,
	134.0
	],
	[
	190.0,
	75.0,
	186.0
	],
	[
	127.0,
	63.0,
	52.0
	],
	[
	141.0,
	182.0,
	25.0
	],
	[
	56.0,
	144.0,
	89.0
	],
	[
	64.0,
	160.0,
	250.0
	],
	[
	182.0,
	86.0,
	245.0
	],
	[
	139.0,
	18.0,
	53.0
	],
	[
	134.0,
	120.0,
	54.0
	],
	[
	49.0,
	165.0,
	42.0
	],
	[
	51.0,
	128.0,
	133.0
	],
	[
	44.0,
	21.0,
	163.0
	],
	[
	232.0,
	93.0,
	193.0
	],
	[
	176.0,
	102.0,
	54.0
	],
	[
	116.0,
	217.0,
	17.0
	],
	[
	54.0,
	209.0,
	150.0
	],
	[
	60.0,
	99.0,
	204.0
	],
	[
	129.0,
	43.0,
	144.0
	],
	[
	252.0,
	100.0,
	106.0
	],
	[
	187.0,
	196.0,
	73.0
	],
	[
	13.0,
	158.0,
	40.0
	],
	[
	52.0,
	122.0,
	152.0
	],
	[
	128.0,
	76.0,
	202.0
	],
	[
	187.0,
	50.0,
	115.0
	],
	[
	180.0,
	141.0,
	71.0
	],
	[
	77.0,
	208.0,
	35.0
	],
	[
	72.0,
	183.0,
	168.0
	],
	[
	97.0,
	99.0,
	203.0
	],
	[
	172.0,
	22.0,
	158.0
	],
	[
	155.0,
	64.0,
	40.0
	],
	[
	118.0,
	159.0,
	30.0
	],
	[
	69.0,
	252.0,
	148.0
	],
	[
	45.0,
	103.0,
	173.0
	],
	[
	111.0,
	38.0,
	149.0
	],
	[
	184.0,
	9.0,
	49.0
	],
	[
	188.0,
	174.0,
	67.0
	],
	[
	53.0,
	206.0,
	53.0
	],
	[
	97.0,
	235.0,
	252.0
	],
	[
	66.0,
	32.0,
	182.0
	],
	[
	236.0,
	114.0,
	195.0
	],
	[
	241.0,
	154.0,
	83.0
	],
	[
	133.0,
	240.0,
	52.0
	],
	[
	16.0,
	205.0,
	144.0
	],
	[
	75.0,
	101.0,
	198.0
	],
	[
	237.0,
	95.0,
	251.0
	],
	[
	191.0,
	52.0,
	49.0
	],
	[
	227.0,
	254.0,
	54.0
	],
	[
	49.0,
	206.0,
	87.0
	],
	[
	48.0,
	113.0,
	150.0
	],
	[
	125.0,
	73.0,
	182.0
	],
	[
	229.0,
	32.0,
	114.0
	],
	[
	158.0,
	119.0,
	28.0
	],
	[
	60.0,
	205.0,
	27.0
	],
	[
	18.0,
	215.0,
	201.0
	],
	[
	79.0,
	76.0,
	153.0
	],
	[
	134.0,
	13.0,
	116.0
	],
	[
	192.0,
	97.0,
	63.0
	],
	[
	108.0,
	163.0,
	18.0
	],
	[
	95.0,
	220.0,
	156.0
	],
	[
	98.0,
	141.0,
	208.0
	],
	[
	144.0,
	19.0,
	193.0
	],
	[
	166.0,
	36.0,
	57.0
	],
	[
	212.0,
	202.0,
	34.0
	],
	[
	23.0,
	206.0,
	34.0
	],
	[
	91.0,
	211.0,
	236.0
	],
	[
	79.0,
	55.0,
	137.0
	],
	[
	182.0,
	19.0,
	117.0
	],
	[
	134.0,
	76.0,
	14.0
	],
	[
	87.0,
	185.0,
	28.0
	],
	[
	82.0,
	224.0,
	187.0
	],
	[
	92.0,
	110.0,
	214.0
	],
	[
	168.0,
	80.0,
	171.0
	],
	[
	197.0,
	63.0,
	51.0
	],
	[
	175.0,
	199.0,
	77.0
	],
	[
	62.0,
	180.0,
	98.0
	],
	[
	8.0,
	91.0,
	150.0
	],
	[
	77.0,
	15.0,
	130.0
	],
	[
	154.0,
	65.0,
	96.0
	],
	[
	197.0,
	152.0,
	11.0
	],
	[
	59.0,
	155.0,
	45.0
	],
	[
	12.0,
	147.0,
	145.0
	],
	[
	54.0,
	35.0,
	219.0
	],
	[
	210.0,
	73.0,
	181.0
	],
	[
	221.0,
	124.0,
	77.0
	],
	[
	149.0,
	214.0,
	66.0
	],
	[
	72.0,
	185.0,
	134.0
	],
	[
	42.0,
	94.0,
	198.0
	]
	]

	assert len(CLASS_NAMES) == len(CLASS_COLORS) == 198