search

megvii-research / CADDM

Official implementation of ID-unaware Deepfake Detection Model
Apache License 2.0
146 stars 21 forks source link

.pt file not found #28

Open t1307109256 opened 11 months ago

t1307109256 commented 11 months ago

The pre-trained model you gave only has a .pkl file but not a .pt file. Where can I get this file? 2023-10-16 18-19-55 的屏幕截图

makaylacmac2 commented 11 months ago

In the file backbones/efficientnet_pytorch/utils.py at the bottom of the file there is code commented out. I assume the first two lines are supposed to be uncommented and the if-else commented out, but it still doesn't work for me. If you figure out this issue, please let me know. image

I was able to download the file manually from the links defined in the mapping variables, but I apparently did not put it in the right location.

t1307109256 commented 11 months ago

May be you can try downloading the .pt files from link https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b3-5fb5a3c3.pth and link https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b4-6ed6700e.pth and then put them in the corresponding path (/data/efficientnet) and see if it works.

molokanov50 commented 7 months ago

@makaylacmac2 @t1307109256 Put a dot before /data since without a dot it is considered as absolute path which obviously doesn't exist in linux. It worked for me.

Gnonymous commented 7 months ago

@makaylacmac2 Sorry to bother you. I would like to ask you if you have successfully performed the training, how was the training dataset's set up, was only one frame extracted from each video, according to extract_frames_idm.py?