1adrianb / face-alignment

:fire: 2D and 3D Face alignment library build using pytorch
https://www.adrianbulat.com
BSD 3-Clause "New" or "Revised" License
6.88k stars 1.33k forks source link

Error in examples/demo.ipynb testing on a batch #332

Open ywangmy opened 1 year ago

ywangmy commented 1 year ago

In "Testing on a batch":

fig = plt.figure(figsize=(10, 5))
for i, pred in enumerate(preds):
    plt.subplot(1, 2, i + 1)
    plt.imshow(frames[1])
    plt.title(f'frame[{i}]')
    for detection in pred:
        plt.scatter(detection[:,0], detection[:,1], 2)

the 2nd loop is redundant, need to plot pred[:, 0], pred[:, 1] only.