Constellation: Learning relational abstractions over objects for compositional imagination