常用的tensorflow函数

在mask_rcnn常用的函数

1  tf.cast():    https://blog.csdn.net/dss875914213/article/details/86558407

2 tf.gather :               https://blog.csdn.net/guotong1988/article/details/53172882 

3 tf.boolean_mask:            https://blog.csdn.net/m0_37393514/article/details/81674489

4  tf.where():                https://blog.csdn.net/ustbbsy/article/details/79564828

5 tf.gather_nd()       多维上进行索引 https://blog.csdn.net/Cyiano/article/details/76087747

a = tf.Variable([[1,2,3,4,5], [6,7,8,9,10], [11,12,13,14,15]])
index_a = tf.Variable([[0,2], [0,4], [2,2]])

print(sess.run(tf.gather_nd(a, index_a)))

# [ 3 5 13]

6 tf.image.crop_and_resize的使用  ROI Pooling层:https://blog.csdn.net/m0_38024332/article/details/81779544

 

posted @   水木清扬  阅读(119)  评论(0编辑  收藏  举报
编辑推荐:
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
点击右上角即可分享
微信分享提示