Open YuXiangLin1234 opened 1 year ago
Hi, I built my code based on https://github.com/biubug6/Pytorch_Retinaface
Hope my answer helps. Thank you!
Thanks for your response, I set weights of different tasks to be 1 and get better results ( I set the weight of loss about gaze to be 10 at the beginning and get bad results).
Hello, I try to reproduce your great work based on the github repo you mentioned before, but results seem to be not good. I'm new to object detection, so there may be some errors in my code. I want to clarify two things:
For downstream 3D gaze head, can I just copy the architecture of location head & landmark head?
class GazeHead(nn.Module): def init(self,inchannels=512,num_anchors=3): super(GazeHead,self).init() self.conv1x1 = nn.Conv2d(inchannels,num_anchors*2,kernel_size=(1,1),stride=1,padding=0) --------------------------> output pitch and yaw