How to generate threebody dataset?

[eric-yyjau]

Hi,

I finished training 'coord_encoder' and move on to generate three-body dataset.
But it looks like the 'file_cnt' pointer is not moving forward.

while file_cnt < 65537:
	generate()
	print(file_cnt)
	if file_cnt % 4096 == 0:
		f = open(f'{outdir}/strokes.txt', 'w')
		f.write(str(dataset))
		f.close()

# output
Starting node.js server...

> agent@0.1.0 start /Users/t_jauy/Documents/strokenet/environment
> node bin/www

<__main__.World object at 0x11d3f0ef0>
0
0

• Generate only one image
  

How could I fix it? Or could I directly download the dataset?
Thank you!

[vexilligera]

Hi,

I tested the script on my local Ubuntu system, got the same output, but the images were correctly generated. If it outputs a bunch of zeros followed by server messages it should be successful, I may need to look at more output messages to confirm the problem.

Unfortunately I didn't keep the dataset. Here's a link to the pretrained models: https://drive.google.com/open?id=1r0Gc_DX-a7XXhNx9diskUikybEMayPDW
However, if you want to finetune or train your own generator since different generators could lead to very different results, I can generate the dataset for you.

[eric-yyjau]

Hi,

Thank you for your reply. I'll try to do further testing.

[seungwonpark]

Thanks for providing pretrained models, @vexilligera !