kanezaki / pytorch-rotationnet

BSD 2-Clause "Simplified" License
80 stars 19 forks source link

Kumoi0728 pose estimation with txt #19

Closed Kumoi0728 closed 2 years ago

Kumoi0728 commented 2 years ago

先生お疲れ様です。Yunyiです。

これは私が書いたpose estimation関連のプログラムです。 このbranchをpushしてみましたが、もしgithubの使え方が間違っていたら教えてください。

前に実行logをチェックしていなかったので、もう一度実行してみたら、test_loaderの長さは1ですが、erumerate(test_loader)はなぜが2回ループしました。1回目のループが終わるとmonitorの画像が得られます。2回目のループではこのようなエラーが出ました:

Traceback (most recent call last): File "pose_estimation.py", line 361, in main() File "pose_estimation.py", line 285, in main pose_estimation(test_loader, model) File "pose_estimation.py", line 320, in pose_estimation s[i num_classes + j] = s[i num_classes + j] scores[vcand[i][k]][k num_classes + j] IndexError: index 839 is out of bounds for axis 0 with size 820

まだ訂正していなくですみません。 書き換えた所に注釈をつけていますので、チェックよろしくお願いします^^。