SSD エラー: ValueError: 配列要素をシーケンスで設定しています (解決済み)

コードはSSD-pytorchコードです

エラーを報告する

 上記のエラー レポートの最後の行にある青いフォントに従って、クリックしてエラーの場所の augmentations.py ファイルに入ります。

 解決

        以前は SSD コードに問題はありませんでしたが、最近 numpy を更新した後、この問題が発生しました: numpy のバージョンが高すぎると、ランダムに選択されたデータの次元が不一致になり、エラーが発生します。

(numpyのバージョンが低い場合、コードは正常に実行できますが、警告が表示されます)

着替える

self.sample_options = np.array([
    # using entire original input image
    None,
    # sample a patch s.t. MIN jaccard w/ obj in .1,.3,.4,.7,.9
    (0.1, None),
    (0.3, None),
    (0.7, None),
    (0.9, None),
    # randomly sample a patch
    (None, None),
], dtype=object)  # 随机选择的数组维度不一致时，需加入dtype=object(numpy版本过高带来的问题)

np.array を使用して配列を生成し、dtype=object を追加して問題を解決します