stylegan2
stylegan2 copied to clipboard
skyflynil
ValueError: Dimension 2 in both shapes must be equal, but are 1024 and 1280.
Hi,
I get this message when I try to run training on my custom dataset.
ValueError: Dimension 2 in both shapes must be equal, but are 1024 and 1280.
I know all my images are 1024x1024 but the error message seems to suggest otherwise. And the same images work fine on the original stylegan2, so not sure where this is going wrong.
Thanks
In the end this worked in Colab for me on 1024x1024 so I guess connected to the min-h and min-w needing to correspond to your images. Hope this helps.
!python run_training.py --num-gpus=1 --data-dir=/yourdatadir --config=config-e --dataset=yourdataset --min-h=8 --min-w=8 --res-log2=7 --mirror-augment=true --metric=none --total-kimg=20000 --result-dir=./results
I am learning stylegan. How do you know that 1024x1024 is --min-h = 8 --min-w = 8 --res- log2 = 7? how is this calculated?
The formula is given in description of this repository: “Create training image set. Instead of image size of 2^n * 2^n, now you can process your image size as of (min_h x 2^n) X (min_w * 2^n) natually. For example, 640x384, min_h = 5, min_w =3, n=7. Please make sure all your raw images are preprocessed to the exact same size. To reduce the training set size, JPEG format is preferred.”
Maybe I have the same problem as you? Can you give me some advice?
Traceback (most recent call last): File "/tensorflow-1.15.2/python3.6/tensorflow_core/python/client/session.py", line 1365, in _do_call return fn(*args) File "/tensorflow-1.15.2/python3.6/tensorflow_core/python/client/session.py", line 1350, in _run_fn target_list, run_metadata) File "/tensorflow-1.15.2/python3.6/tensorflow_core/python/client/session.py", line 1443, in _call_tf_sessionrun run_metadata) tensorflow.python.framework.errors_impl.InvalidArgumentError: 2 root error(s) found. (0) Invalid argument: Cannot batch tensors with different shapes in component 0. First element had shape [3,1024,1024] and element 1 had shape [3,3840,2699]. [[{{node GPU0/DataFetch/IteratorGetNext}}]] (1) Invalid argument: Cannot batch tensors with different shapes in component 0. First element had shape [3,1024,1024] and element 1 had shape [3,3840,2699]. [[{{node GPU0/DataFetch/IteratorGetNext}}]] [[GPU0/DataFetch/IteratorGetNext/_5073]] 0 successful operations. 0 derived errors ignored.
