How to change number of elements?

[sukritiverma1996]

Hey,
I've been trying to train a model on Rico with more data
In rico.py, I can see that number of elements are capped at 9

I've been going through your generator and discriminator code and I would greatly appreciate some help in where I need to make changes so the models can accept in data samples with bboxes and labels > 9.

Thanks so much!
Sukriti

[ktrk115]

Hi Sukriti,

If I remember correctly, you don't need to edit the code for both the generator and the discriminator. The code should work with variable length of elements. To add more labels, you need to edit the following.

const_layout/data/rico.py

Lines 19 to 33 in 4090f55

	labels = [
	'Toolbar',
	'Image',
	'Text',
	'Icon',
	'Text Button',
	'Input',
	'List Item',
	'Advertisement',
	'Pager Indicator',
	'Web View',
	'Background Image',
	'Drawer',
	'Modal',
	]

If you get any results with the full Rico dataset, I would appreciate it if you could share them here or on PR.

Thanks!
Kotaro

[sukritiverma1996]

Yes, definitely
This is my first time dabbling with GANs though
Thank you!