aprillysea

Traceback (most recent call last): File "E:\code\PFLD-pytorch-master\train.py", line 221, in main(args) File "E:\code\PFLD-pytorch-master\train.py", line 146, in main optimizer, epoch) File "E:\code\PFLD-pytorch-master\train.py", line 61, in train features, landmarks = pfld_backbone(img) File...

I have trained the model,but the result is not good.it seems like the loss's problem.

models

1

Traceback (most recent call last): File "visualize/generate.py", line 26, in from models.ce import make_model ModuleNotFoundError: No module named 'models'

RuntimeError: Given groups=1, weight of size [10, 3, 3, 3], expected input[1, 361, 424, 3] to have 3 channels, but got 361 channels instead

train dataset

1

Can the author provide a training dataset? especially in numpy format