Label code - Githubissues

LymanSong / FP_GNN

A framework for detecting and classifying indoor elements in a floor plan using Graph Neural Network.

MIT License

20 stars 4 forks source link

Label code #2

Open JanineCHEN opened 3 years ago

JanineCHEN commented 3 years ago

It seems like there are 8 possible labels for each polygon, yet I could not find the list of names for these 8 labels, namely which one corresponding to what category, wall, door or room?

Could you kindly provide this list? Thanks a bunch!

LymanSong commented 3 years ago

For Cubicasa dataset the labels are:

class 0(objects)
class 1(wall)
class 2(window)
class 3(door)
class 4(stair)
class 5(room)
class 6(porch)
class 7(outer space)

For UOS dataset,

class 0(objects)
class 1(wall)
class 2(window)
class 3(door)
class 4(stair)
class 5(lift)
class 6(room)
class 7(hallway)
class 8(X-room)

For more details, please see the paper [https://www.preprints.org/manuscript/202101.0584/v1]

Thank you.