訓練済みYOLOXでサクッと推論したい場合に使える方法
89{icon} {views} 物体検出を使いたい場合、カスタムデータで訓練することなく、COCO Pretrainedモデルによる推論だけで事足りることも多々あります。2021年最新モデルのYOLOXを使って、サクッ […]...
論文まとめ:Deep Spectral Methods: A Surprisingly Strong Baseline for Unsupervised Semantic Segmentation and Localization
44{icon} {views} タイトル:Deep Spectral Methods: A Surprisingly Strong Baseline for Unsupervised Semantic Segmenta […]...
PyTorch lightningのCSVLoggerを攻略する
48{icon} {views} PyTorch lightningのロガーとしてTensorBoardがデフォルトですが、出てきた評価指標を解析するとCSVでロギングできたほうが便利なことがあります。lightning […]...
PyTorch LightningのGPU訓練で「RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same」と怒られた場合
27{icon} {views} PyTorch LightningはGPUで訓練しても「.to(device)」のようなデバイス指定をしなくていいのが売りですが、「デバイスが異なる」と怒られてハマってしまったので、解決 […]...
論文まとめ:Discovering Underground Maps from Fashion
58{icon} {views} タイトル:Discovering Underground Maps from Fashion 論文:https://arxiv.org/abs/2012.02897 https://ar […]...
論文まとめ:(DALL・E 2論文)Hierarchical Text-Conditional Image Generation with CLIP Latents
312{icon} {views} タイトル:Hierarchical Text-Conditional Image Generation with CLIP Latents リンク:https://cdn.openai […]...
論文まとめ:One-Shot Adaptation of GAN in Just One CLIP
149{icon} {views} 引用 タイトル:One-Shot Adaptation of GAN in Just One CLIP 論文:https://arxiv.org/abs/2203.09301 コード: […]...
PyTorch lightningで訓練時間を指定して自動的に訓練終了+再開(Google Colab向け)
179{icon} {views} PyTorch lightningでは、Trainerにオプションを追加するだけで、訓練時間を指定できます。これとCheckpointを組み合わせるとGoogle Colabでかなり便 […]...
argparseのハイフン置き換え仕様でハマった
102{icon} {views} argparseではハイフンのコマンドライン引数をアンダーバーに置き換える仕様があります。これはPythonの言語仕様の都合によるものですが、思わぬハマり方をしたので自分のメモ用に書い […]...
PyTorch→ONNXのコンバートでモデルの入力サイズを可変にして推論する
382{icon} {views} ONNXでモデルは入力サイズを可変にできます。PyTorch→ONNXの変換と、ONNXRuntimeでの推論方法、また可変にしたことによる速度の副作用を検証していきます。 きっかけ […]...