Questions about the "get_class_embedding.py" file

[zj916716524]

Hello. I came across your paper and found it very interesting, so I downloaded the code to run it myself. And I want to make my own dataset, I have modified the paths_config.py files. When I run the get_class_embedding.py file, unfortunately it gives me this error "ImportError: No module named 'fused'"

[UniBester]

Could you provide the detail of your error?

[zj916716524]

Thank you very much for your reply, I suspect that I am having problems making my own dataset, I am downloading the dataset of flowers you provided this morning and I plan to make my own dataset after running the dataset you gave me as a reference first. I'll come back chatter if I don't solve the problem.

[zj916716524]

I downloaded the dataset for flowers and created the packet files with "all_img/test" and "all_img/train", and modifications the config/paths_config.py and config/data_config.py

config/paths_config.py

dataset_paths = { 'af_train': '/home/zeng/AGE-main/all_img/train', 'af_valid': '/home/zeng/AGE-main/all_img/test', },

config/data_config.py

from configs import transforms_config from configs.paths_config import dataset_paths DATASETS = { 'af_encode': { 'transforms': transforms_config.EncodeTransforms, 'train_source_root': dataset_paths['af_train'], 'train_target_root': dataset_paths['af_train'], 'valid_source_root': dataset_paths['af_valid'], 'valid_target_root': dataset_paths['af_valid'], }, }

The program has reported an error

Traceback (most recent call last): File "tools/get_class_embedding.py", line 19, in from models.age import AGE File "./models/age.py", line 12, in from models.encoders import psp_encoders File "./models/encoders/psp_encoders.py", line 8, in from models.stylegan2.model import EqualLinear File "./models/stylegan2/model.py", line 7, in from models.stylegan2.op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d File "./models/stylegan2/op/init.py", line 1, in from .fused_act import FusedLeakyReLU, fused_leaky_relu File "./models/stylegan2/op/fused_act.py", line 9, in fused = load( File "/home/zeng/anaconda3/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1079, in load return _jit_compile( File "/home/zeng/anaconda3/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1317, in _jit_compile return _import_module_from_library(name, build_directory, is_python_module) File "/home/zeng/anaconda3/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1699, in _import_module_from_library file, path, description = imp.find_module(module_name, [path]) File "/home/zeng/anaconda3/lib/python3.8/imp.py", line 296, in find_module raise ImportError(_ERR_MSG.format(name), name=name) ImportError: No module named 'fused'

I found that the naming you used in the given dataset is n02085620_25.JPEG_238_24_392_167.jpg, but the data I downloaded from the internet is named image_00033.jpg, did you perform any manipulation in between? Like stitching the original image and the target like zi2zi?

[UniBester]

I guess the program error is because there is something wrong with your environment configuration. You can configure your environment using environment/envrionment.yaml.

As for the naming problem, you should rename you data file in the form of catefgory-id_sample_id.jpg. For example, in n02085620_25.JPEG_238_24_392_167.jpg, "n02085620" is the category id and "25.JPEG_238_24_392_167" is the sample id. They are connected by a "_".

[fb-reps]

i down your weight in face dataset,but the opt has no attribution named A_length. the error information is:

Traceback (most recent call last): File "/home/user/.pycharm_helpers/pydev/pydevd.py", line 1483, in _exec pydev_imports.execfile(file, globals, locals) # execute the script File "/home/user/.pycharm_helpers/pydev/_pydev_imps/_pydev_execfile.py", line 18, in execfile exec(compile(contents+"\n", file, 'exec'), glob, loc) File "/home/user/Document/fanbing/AGE/tools/get_class_embedding.py", line 82, in run() File "/home/user/Document/fanbing/AGE/tools/get_class_embedding.py", line 41, in run net = AGE(opts) File "../models/age.py", line 81, in init self.ax = Ax(self.opts.A_length) AttributeError: 'Namespace' object has no attribute 'A_length'