Pipeline.config でクラスを 1 に設定する必要があります。クラスの数を 1 に変更し、エラーを表示するトレーニングを開始しました。
InvalidArgumentError (see above for traceback): Restoring from checkpoint failed.
これは、現在のグラフとチェックポイントからのグラフの不一致が原因である可能性が最も高いです。
チェックポイントに基づいて予想されるグラフを変更していないことを確認してください。元のエラー:
Assign requires shapes of both tensors to match. lhs shape= [1,1,256,12] rhs shape= [1,1,256,546]
[[node save/Assign_86 (defined at /home/drive/Downloads/models-
master/research/object_detection/legacy/trainer.py:378) @bmabey @alextp
@moonboots @ewilderj @jtoy @tensorflowbutler
また、微調整チェックポイントを設定する方法は?