whydna / RoomNet

Training diffusion model to generate furnished room -> empty room (and vice versa).

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

RoomNet

Training a ControlNet to generate furnished room -> empty room (and vice versa).

Example 1

Source: Screenshot 2024-03-15 at 10 18 12 AM

Target: Screenshot 2024-03-15 at 10 21 32 AM

Model Result: Screenshot 2024-03-15 at 10 17 59 AM

Example 2

Source: Screenshot 2024-03-15 at 10 20 30 AM

Target: Screenshot 2024-03-15 at 10 21 54 AM

Model Result: Screenshot 2024-03-15 at 10 20 41 AM

First Training Run

Screenshot 2024-03-15 at 10 29 44 AM

Second Training Run

Screenshot 2024-03-15 at 10 30 41 AM

About

Training diffusion model to generate furnished room -> empty room (and vice versa).


Languages

Language:Python 99.9%Language:Shell 0.1%