VisionRush / DeepFakeDefenders

Image forgery recognition algorithm
Apache License 2.0
586 stars 73 forks source link

missing final_model_csv and pre_model dir #2

Open JackGit opened 2 months ago

JackGit commented 2 months ago

missing final_model_csv and pre_model dir while building docker from Dockerfile

XinyiZhangcasia commented 2 months ago

./final_model_csv is where we store our final models. Please remove the command for creating this folder when building docker.

Before running the training, please make sure to download the ImageNet-1K pre-trained weights for the two models from the following links and place them in the./pre_modeldirectory:

RepLKNet: https://drive.google.com/file/d/1vo-P3XB6mRLUeDzmgv90dOu73uCeLfZN/view?usp=sharing ConvNeXt: https://dl.fbaipublicfiles.com/convnext/convnext_base_1k_384.pth

We will update README.md soon.

jh-hub-create commented 2 months ago

XinyiZhangcasia 评论 2024 年 9 月 4

https://github.com/VisionRush/DeepFakeDefenders/issues/2#issuecomment-2328176017