search

NeonLeexiang / DLGSANet

Apache License 2.0
86 stars 8 forks source link

Problem with Basic_SR_inference_DFDNET and a question about DFDNET #6

Open BeauGeogeo opened 1 year ago

BeauGeogeo commented 1 year ago

Good morning,

First of all thank you for sharing this great work ! I'd like to use it for a talking head project of the company I'm currently working for.

I tried to run the notebook for Basic SR inference but got the following error :

/usr/local/lib/python3.10/dist-packages/torchvision/transforms/functional_tensor.py:5: UserWarning: The torchvision.transforms.functional_tensor module is deprecated in 0.15 and will be removed in 0.17. Please don't rely on it. You probably just need to use APIs in torchvision.transforms.functional or in torchvision.transforms.v2.functional. warnings.warn( Please install facexlib: pip install facexlib Traceback (most recent call last): File "/content/BasicSR/inference/inference_dfdnet.py", line 114, in net = DFDNet(64, dict_path=args.dict_path).to(device) File "/content/BasicSR/basicsr/archs/dfdnet_arch.py", line 77, in init self.dict = torch.load(dict_path) File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 815, in load return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args) File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1033, in _legacy_load magic_number = pickle_module.load(f, pickle_load_args) _pickle.UnpicklingError: invalid load key, '<'.

I also wanted to ask a question. You have scripts for DFDNET inference, why ? And where is exactly the script to test your own model DGLSANet ?

Thank you very much.

Best regards,

Geoffrey.

NeonLeexiang commented 1 year ago

Sorry for being late with the replies, the released code is simply based on the BasicSR framework, and the proposes of the released code is to offer the arch file of the main network structure of the DLGSANet.

We are sorry that we only offer the arch file and options file of the network structure with the pre-train models and some visual results.

NeonLeexiang commented 1 year ago

Sorry for being late with the replies, the released code is simply based on the BasicSR framework, and the proposes of the released code is to offer the arch file of the main network structure of the DLGSANet.

We are sorry that we only offer the arch file and options file of the network structure with the pre-train models and some visual results.

Given that the official BasicSR framework provides the inference of several networks, I'll put it on my to-do list and get it done shortly.