xmy0916 / siamesenet_face_compare Goto Github PK

Q:1
I wish the author add if __name__=='__main__': sys.exit(main())
Q:2
File "F:/temp/SiameseNet_face_compare-master/SiameseNet_face_compare-master/train.py", line 65, in <module> sys.exit(main()) File "F:/temp/SiameseNet_face_compare-master/SiameseNet_face_compare-master/train.py", line 52, in main output1,output2 = net(img0,img1) File "C:\Users\PHILIPS\AppData\Roaming\Python\Python36\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl return forward_call(*input, **kwargs) File "F:\temp\SiameseNet_face_compare-master\SiameseNet_face_compare-master\model.py", line 64, in forward output1 = self.forward_once(input1) File "F:\temp\SiameseNet_face_compare-master\SiameseNet_face_compare-master\model.py", line 57, in forward_once draw_np = output.cpu().detach().reshape([8,1,100,100]) RuntimeError: shape '[8, 1, 100, 100]' is invalid for input of size 5120000
can you help me ?