2020-02-01から1ヶ月間の記事一覧

ONNXモデルの中間ノード出力を抽出するライブラリを作ってみた

先日記事にしたこれを実装してみた。 maminus.hatenadiary.org ソースコード GitHubに上げました。まともな?ライブラリを作るの初めてなので何か間違ってるかも・・・(特にPipenv自信なし。pip install も直接gitを指定できなかった。たぶんどこか間違って…

浮動小数点誤差の自分メモ

普段適当にfloat32やfloat16使ってるけど、適当すぎてたまに困るので勉強のために机上での考察とプログラムを動かして検証してみた。 オーバーフロー、アンダーフロー 指数部が表現可能な範囲外を超えるケース float32なら指数が+127を超えるとオーバーフロ…

ONNXモデルの中間層出力を取得するアイデア

忘れないように自分用のメモ。 やりたいこと ONNXモデルを使って推論をする時に最終出力ではなく中間ノードの出力がほしい アイデア 2フェーズに分けて処理する 第一フェーズでは中間ノードの出力にShapeノード+出力ノードを接続する make_node('Shape', i…

OpenCL/CUDAのスレッド間通信めも

CUDAとOpenCLでスレッド同士の通信でどうするんだっけ?と思って調べたことを自分用にメモ。 AMD(OpenCL)はshuffleがなさそう retval = sub_group_broadcast(value, thread_id)がある 特定の1つのthread_idからWavefront全体に値をわたせる 隣のスレッドと…

YUY2のRAW画像を保存してPythonでRGB化

やったこと YUY2出力可能なUSBカメラをLinuxに接続 ffmpegでRAW画像キャプチャしてファイル保存 Pythonからnumpyで読み込み+OpenCVでRGBへ変換 正確にはBGRにしてファイル保存 キャプチャ(コマンドライン) $ ffmpeg -f v4l2 -input_format yuyv422 -video…