Pytorchで割とよく検索するけど、ひとつの記事にするほどでもない事をまとめてみる。使っていて便利そうだと思った情報はどんどん追加していくつもり。 cuda.is_available(): GPU動作チェック GPUかCPUかを自動判断 to(x.deice): model中での変数がGPUかCPU…
話題のEfficientNetを実装してみる。基本的な構造はNASNetとほぼ変わらないんだけど、EfficientNet特有の広さ、深さ、解像度などのパラメータも含めてコードを書いてみる。 画像はこちらのサイトから引用しました。 環境 python 3.7.4 torch 1.0.0 ヘッダ im…
画素毎にテクスチャの特徴抽出をするのに、全ての画素に対してその周辺窓のGLCMを計算したい。この計算をするのに、pythonなのに画素の二重forループ使って計算する方法があるけど、遅すぎて使い物にならない。そこで、numpyを使って画素毎のGLCM計算をスク…
転移学習、スタイル変換、物体検知、セマンティックセグメンテーション、メトリックラーニング、perceptual loss、ゼロショット学習など学習済みモデルの中間層を使いたい場合がよくある。Pytorchで使える学習済みモデルの特徴マップと特徴ベクトルを抽出す…
OpenCVなどでUSBカメラなどから得られた動画を流している際に、各フレームの解析結果をグラフにして動画と一緒にリアルタイムで表示したいときがある。このようなときにプロット図をいったんファイルに保存しながらやると面倒なので、できればメモリ上でのや…
Pytorchを使って1次元の線形回帰 (Linear Regression) の学習を実装してみる。目的はPytorchのモデルの書き方や学習コードの書き方の再確認。 線形回帰モデル まずはPytorchで線形回帰モデルを書いてみる。パラメータは傾き(weight)と切片(bias)の2つ from…
pythonのバージョンやフォルダの確認、pipのバージョン、pipでインストールできるライブラリのバージョンなどを確認する方法をリストアップしてみる。この記事で確認できるのは以下の情報。 pythonそのもののバージョン pipそのもののバージョン pipでインス…