keras-ctpn icon indicating copy to clipboard operation
keras-ctpn copied to clipboard

Published 20 hours ago • verified publisher icon yizt

关于度量的数据类型

Open RabbearSu opened this issue 5 years ago • 4 comments

https://github.com/yizt/keras-ctpn/blob/22bdee3a5a60baa29f9bc51d73e297c199547516/ctpn/layers/target.py#L204 在前面ctpn_target_graph的输出中，后三个度量的类型均为tf.float32，可是这里倒数第四个指定类型为tf.int64，是否应为tf.float32呢？

RabbearSu avatar Apr 19 '19 03:04 RabbearSu

@RabbearSu 这个代码已经注释了哦

yizt avatar Apr 19 '19 03:04 yizt

@yizt 我看您在tf_utils.py里备注说尽量用原生的tf.map_fn，效率更高

RabbearSu avatar Apr 19 '19 05:04 RabbearSu

@RabbearSu 一般情况下用tf.map_fn是更好一些，我这里加了自定义度量后，还用tf.map_fn遇到了bug，没有解决；所以还是改为用batch_slice了。

yizt avatar Apr 19 '19 07:04 yizt

@yizt 原来是这样，我尝试一下能不能改成tf.map_fn

RabbearSu avatar Apr 20 '19 02:04 RabbearSu