有没有图像描述生成器代码

[Dmygithub]

你好，模型训练好后应该如何应用呢？
例如给一张图片，如何生成图像描述？

[zchoi]

您好，由于本项目是离线视觉特征训练和测试，如果要实际部署，需要修改成端到端推理。有几点建议包括但不限于：1. 修改dataloader，从而对输入图像进行online的特征抽取，见#12 ，2. 下面的代码包含了模型输出解码和评测，gts如果不需要你可以删除掉，只要模型当前的输出gen_i即可。

S2-Transformer/test_transformer.py

Lines 33 to 37 in c2f56c5

	caps_gen = text_field.decode(out, join_words=False)
	for i, (gts_i, gen_i) in enumerate(zip(caps_gt, caps_gen)):
	gen_i = ' '.join([k for k, g in itertools.groupby(gen_i)])
	gen['%d_%d' % (it, i)] = [gen_i.strip(), ]
	gts['%d_%d' % (it, i)] = gts_i