single-stage-pose
single-stage-pose copied to clipboard
3D Pose Augmentation
Hi, I am trying to merge the PVNet code with your single stage code. As part of training, the images undergo augmentation such as rotation, cropping and scaling. I have applied the corresponding augmentation to the 3D pose but the network is very slow to train. After 120 epochs, it is performing very poorly even on train data. Removing augmentation altogether causes the network to overfit. It would be great if you could suggest where I am going wrong. Thanks!