How can i add more more objects to the sketch ?

noyoshi / smart-sketch

🖌 photorealistic drawings from simple sketches using NVIDIA's GauGAN

GNU General Public License v3.0

233 stars 35 forks source link

How can i add more more objects to the sketch ? #9

Open parkzinc opened 5 years ago

parkzinc commented 5 years ago

Its possible add more like animals, etc ?

noyoshi commented 5 years ago

Yeah, you can. Just need to add label mapping that map the color the user uses, to the color the model is expecting. You can go to the NVIDIA repo and find the COCO stuff dataset greyscale colors that they associate with each class.