Skip to content

Commit

Permalink
fix bugs with LLM prompt
Browse files Browse the repository at this point in the history
  • Loading branch information
atwang16 committed Jul 14, 2024
1 parent 2f3e2c8 commit 6304c1a
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion libsg/model/sg_parser/llm_sg_parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,8 +57,14 @@
Requirements:
The object and relationship IDs should start with 0 and increment. Every subject_id and target_id in relationships should correspond to an existing object ID.
If the room type is bedroom, the object name must be one of "armchair", "bookcase", "cabinet", "ceiling lamp", "chair", "children cabinet", "coffee table", "desk", "double bed", "dressing chair", "dressing table", "kids bed", "nightstand", "pendant lamp", "shelf", "single bed", "sofa", "stool", "table", "tv stand", or "wardrobe".
If the room type is diningroom or livingroom, the object name must be one of "armchair", "bookcase", "cabinet", "ceiling lamp", "chaise longue sofa", "chinese chair", "coffee table", "console table", "corner/side table", "desk", "dining chair", "dining table", "l-shaped sofa", "lazy sofa", "lounge chair", "loveseat sofa", "multi-seat sofa", "pendant lamp", "round end table", "shelf", "stool", "tv stand", "wardrobe", or "wine cabinet".
Ensure to include common essential objects for the room type, even if not explicitly mentioned in the input text.
The scene graph should have a minimum of 5 objects for any room type, if not explicitly mentioned.
The scene graph should have a minimum of 5 objects for any room type, if not explicitly mentioned. These objects should
be reasonably spatially related to the existing objects.
If a number of objects are specified, please include each object in the count as a separate node. For example, if the text specifies "two chairs", include two separate nodes for the chairs.
Expand Down

0 comments on commit 6304c1a

Please sign in to comment.