Pytorch

【深層距離学習】画像分類が難しい場合の学習法

前書き 画像分類をする上でどうしても特徴が2クラス間で似通っており、 分類させるのが難しいといったケースが存在します。 そういった場合にどういった学習が有効なのか調査してみました。 深層距離学習 深層距離学習の考え方についてはこちらの記事を参考…

【Pytorch】MNISTで簡単なアンサンブル学習を実装

はじめに Kaggleなどを勉強していて、 不均衡データが発生した時にどうすりゃいいんだとなる事があったので、 アンサンブル学習的な要素を含めて実装していきます。 データの不均衡 片方のデータの取得が困難な場合に発生します。 異常検知では、そもそも異…

【Pytorch】CUDA error: device-side assert triggered で発生したエラー

エラー CUDA error: device-side assert triggered ビビりました。 単刀直入によるとネットワークに入力したデータの形式が間違っている場合に、 発生するエラーだそうです。 やろうとしたこと CIFAR10の画像から2クラスのみを取得して、二値分類を試そうと…

【U-Net】PytorchでSemantic Segmentationの推論+切り抜きを実装

前書き melheaven.hatenadiary.jp前回、上記の記事のように学習を進めました。 今回は推論をやっていきます。 推論の準備 学習後、作成した重みファイル(.pth)をロードします。 # 学習モデルを引っ張ってくる(前回の記事参照) model = ResNetUNet(num_clas…

【U-Net】Semantic Segmentationの学習フェーズの実装

前書き Kaggle(やってない)や研究の勉強ついでに。 ネットワークモデルは自分で構築するより、 外から取ってきて自分の実現したい内容に合わせるかが重要です。 概要 前回dataloaderの作成まで行いました。 今回はU-Netでmelheaven.hatenadiary.jp今回はU-…

Pytorchのデータ前処理【Dataset・Transform・DataLoader】を実装

前書き 今までTensorflowを活用していたのですが、toPytorchを勉強しています。 今日は基礎をざっと紹介していきます。melheaven.hatenadiary.jp Pytorchの機能 torch → Tensorの作成や操作 torch.Tensor → torch.Tensorクラスのパッケージ化 torch.autograd…

Keras(on Tensorflow) vs Pytorch

前書き 最近研究でKerasとPytorchを対象研究ごとに使い分けたりすることが多いのですが、 その比較がイマイチ曖昧でしたのでまとめることにしました。 KerasもPytorchも大きな変革が・・・ 2019年10月、KerasとPytorchに大きな変革がもたらされました。 Kera…