Example

采用 layout-generator 生成数据集，利用 FPN 模型进行 layout 到 heat_field 的回归

Usage

• 查看参数帮助 python train.py -h

• 使用配置文件 config.yml 进行训练。注意参数data_root 为数据集目录，需要包含子目录 train 与 test

  python train.py --config config.yml
  

• 训练过程中默认打印模型参数量，采用 tensorboard 记录 hparams，train_loss，val_loss，model 的输入与输出图像

• tensorboard --logdir . 查看结果

Result

• 160 train，40 validation
• 参数

k e y	c o n f i g	resu me_f rom_ chec kpoi nt	s e e d	g p u s	u s e _ 1 6 b i t	val _ch eck _in ter val	t e s t _ a r g s	da ta _r oo t	t r a i n _ s i z e	d r o p _ p r o b	i n p u t _ s i z e	m e a n _ l a y o u t	s t d _ l a y o u t	m e a n _ h e a t	s t d _ h e a t	m a x _ e p o c h s	o p t i m i z e r _ n a m e	l r	b a t c h _ s i z e	n u m _ w o r k e r s
v a l u e	c o n f i g . y m l	None	1	1	F A L S E	1	F A L S E	d: /w or k/ da ta se t1	8	2	2 0 0	5 0 0 0	1 0 0 0 0	2 9 8	2	2 0 0	a d a m	0 0 1	6 4	4

• loss

• image