rpeng666 / facedetection_with_detr Goto Github PK

Thank you for your sharing

I tried to replicate your project

However, after I restored the environment on my computer, I got an error：

Traceback (most recent call last):
File "D:/face_detection/face_detection_detr/FaceDetection_With_DETR-main/main.py", line 87, in
main()
File "D:/face_detection/face_detection_detr/FaceDetection_With_DETR-main/main.py", line 28, in main
model.load_state_dict(torch.load('./model.pth'))
File "D:\anaconda\envs\Pytorch\lib\site-packages\torch\serialization.py", line 579, in load
with _open_file_like(f, 'rb') as opened_file:
File "D:\anaconda\envs\Pytorch\lib\site-packages\torch\serialization.py", line 230, in _open_file_like
return _open_file(name_or_buffer, mode)
File "D:\anaconda\envs\Pytorch\lib\site-packages\torch\serialization.py", line 211, in init
super(_open_file, self).init(open(name, mode))
FileNotFoundError: [Errno 2] No such file or directory: './model.pth'

I can't find the file ''model.pth "in the downloaded ZIP file.

I don't know how to get this file (API？Link？)

Looking forward to your reply!