問題タブ [torchvision]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
python - Pytorch: 参照テーブルを使用してカスタム データセットを作成する方法
タイプ、クラス、パスの 3 つの列を持つ reference.csv ファイルがあります。最初の 5 行の例を次に示します。
より見やすい形式で:
DataLoader (torch.utils.data.DataLoader) を使用できるように、データセット クラス (torch.utils.data.Dataset) を作成して画像を読み取りたいと考えています。
参照テーブルを使用してカスタム データセットを作成する正しい方法は何ですか?
machine-learning - PyTorch - BCELoss: ValueError: ターゲットと入力には同じ数の要素が必要です
BCELoss
ニューラル ネットワークの損失関数として使用すると、 ValueError: Target and input must have the same number of elements
.
これがテスト段階の私のコードです(これは非常に典型的なテスト段階のコードです):
変数の形状は、 ( MNISTデータセット内の) ターゲット クラスoutput
が[1000, 10]
存在するためであり、変数の形状は、テスト対象のバッチのターゲット クラスが含まれているためです (テストのバッチ サイズは に設定されています)。では、問題は、ネットワークの損失関数としてどのように適用できるかということです。10
target
[1000]
10
BCELoss
CNN
ps 私が使用するデータセットは、ライブラリによって提供されるMNISTtorchvision
データセットです。
psここで同様の質問に対する回答は、私の場合の解決策を提案していません。