Closed Arvohy closed 2 years ago
Please check if the pose data prepared correctly first.
Hi, I checked the dataroot which was the same as you described in README.
Then I printed the keypoints and pose in the function def _load_kpt(self, name):
( in the deepfashion_datasets.py
file).
keypoints_y [41, 87, 85, 146, 205, 90, 156, 216, 210, -1, ...
keypoints_x [103, 87, 45, 26, 36, 129, 137, 132, 61, -1, -...
Name: fashionMENSweatersid0000085901_1front.jpg, dtype: object
[[ 41 103]
[ 87 87]
[ 85 45]
[146 26]
[205 36]
[ 90 129]
[156 137]
[216 132]
[210 61]
[ -1 -1]
[ -1 -1]
[215 107]
[ -1 -1]
[ -1 -1]
[ 32 94]
[ 32 109]
[ 34 75]
[ -1 -1]]
[[[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
1.2529010e-41 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
9.9231549e-41 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
7.6442933e-40 0.0000000e+00]
...
[2.0150111e-40 0.0000000e+00 0.0000000e+00 ... 2.1813527e-32
0.0000000e+00 0.0000000e+00]
[2.8430945e-41 0.0000000e+00 0.0000000e+00 ... 3.5363172e-33
0.0000000e+00 0.0000000e+00]
[3.9012149e-42 0.0000000e+00 0.0000000e+00 ... 5.5758720e-34
0.0000000e+00 0.0000000e+00]]
[[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
3.1771640e-41 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
2.5164518e-40 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
1.9385297e-39 0.0000000e+00]
...
[6.2066732e-40 0.0000000e+00 0.0000000e+00 ... 5.2327930e-32
0.0000000e+00 0.0000000e+00]
[8.7572746e-41 0.0000000e+00 0.0000000e+00 ... 8.4831842e-33
0.0000000e+00 0.0000000e+00]
[1.2017536e-41 0.0000000e+00 0.0000000e+00 ... 1.3375822e-33
0.0000000e+00 0.0000000e+00]]
[[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
7.8363413e-41 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
6.2066732e-40 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
4.7812766e-39 0.0000000e+00]
...
[1.8594166e-39 0.0000000e+00 0.0000000e+00 ... 1.2208928e-31
0.0000000e+00 0.0000000e+00]
[2.6235110e-40 0.0000000e+00 0.0000000e+00 ... 1.9792599e-32
0.0000000e+00 0.0000000e+00]
[3.6002160e-41 0.0000000e+00 0.0000000e+00 ... 3.1207891e-33
0.0000000e+00 0.0000000e+00]]
...
[[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
...
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]
[[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
...
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]
[[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
...
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]]
Is there something wrong with pose calculation function? Or the data I put isn't correct.
so it is a heatmap in 18x256x176? what makes it "most invalid"?
Yeah, it's a heatmap in 18×256×176.
Then I commented out the line if pose != None:
, and it worked. ( I still don't know why it got Nonetype. )
Thanks for your help and patience.🌹
Hi Aiyu Cui, When I was running the pose transfer part of demo.ipynb file, I found something wrong with the pose tensor. (I finished the training process and no error occured.)
Then I printed the 'pose', and found it was mostly invalid.
Sorry for distrubing you, and look forward to your suggestion.