随笔分类 - AI part
摘要:1. tf.keras.layers.Reshape会变成一个layer, 而tf.reshape只是一个计算;用tf.reshape训练假设得到100层的模型,tf.keras.reshape训练五层会有105层模型; 2. tf.keras.layers.Reshape会消耗更多的算力和内存;
阅读全文
摘要:先简单记录版本,以后再详细; haotian-liu/LLaVA at v1.2.1 (github.com) 当前调通版本: cuda11.7; cudnn LLava-1.2.1; python3.10; torch2.0.1 flash-attn; flash_attn-2.0.7+cu117
阅读全文
摘要:做模型部署边缘设备的时候,我们经常会遇到特定格式的要求。但常见的onnx2tf很多时候都不能满足我们的要求。因此,记录一下我的操作过程。 1. 环境:(linux18.04) # Name Version Build Channel _libgcc_mutex 0.1 main defaults _
阅读全文
摘要:背景如下:我要删除Round节点, 同时看了一下,Dequantize和Quantize也是没有必要的。所以最好一起删除。 原始项目地址:PINTO0309/hand-gesture-recognition-using-onnx: This is a hand gesture recognition
阅读全文
摘要:踩坑: 将slim和models路径加入虚拟环境中; 将slim和models路径加入系统路径中;(加入系统路径的时候,这个pwd) qit (持续更新) 具体安装流程: (很多问题)2.0版本 ;(继续更新) conda create -n tf_obj_det_api_v2 python=3.8
阅读全文
摘要:上述为按照yolo v8 pose官方教程写的代码。 使用了python3.7, 以及对应的torch版本如下后,报错。AttributeError: module ‘torch‘ has no attribute ‘_six‘ 解决方法:更换了虚拟环境,重新安装yltralytics. 新环境:
阅读全文