jarr0d / vitaa Goto Github PK

I guess to_bgr255 is used for Caffe2. This should be set to False.

When I run train_net, error occurs like:
return F.conv2d(input, weight, self.bias, self.stride,
RuntimeError: CUDA out of memory. Tried to allocate 12.00 MiB (GPU 0; 7.79 GiB total capacity; 6.48 GiB already allocated; 11.12 MiB free; 6.52 GiB reserved in total by PyTorch)
I was wondering whether the error is the one that my GPU is not supported by batch_size=64 as your mentioned. My GPU is 2080 with 16G memory. How can I fix it, is any configuration instruction I can use?
Thanx for your help.

when I run 'python tools/train_net.py --config-file configs/cuhkpedes/bilstm_r50_seg.yaml', The following errors occurred:
Traceback (most recent call last):
File "tools/train_net.py", line 23, in
from vitaa.data import make_data_loader
File "./vitaa/data/init.py", line 2, in
from .build import make_data_loader
File "./vitaa/data/build.py", line 7, in
from . import datasets as D
ImportError: cannot import name 'datasets'
According to these problems, I can't find datasets in "./vitaa/data"
How to solve this problem?

@Jarr0d hi thanks for opesourcing the code based , can you please share ur pretrained weights on google drive or one drive

Thanks in adavance

When I entered this command called “python -m torch.distributed.launch --nproc_per_node=4 tools/train_net.py --config-file configs/cuhkpedes/bilstm_r50_seg.yaml ”, The following error occurred:
subprocess.CalledProcessError: Command '['/home/whh/anaconda3/envs/hpn/bin/python', '-u', 'tools/train_net.py', '--local_rank=0', '--config-file', 'configs/cuhkpedes/bilstm_r50_seg.yaml']' returned non-zero exit status 1.

This problem has not been solved yet. What command should I enter to conduct multi-gpu training?

Hello, the download link of the file "text_attribute_graph" is invalid. Is there any new link? Or please send a copy to my email: [email protected]. Thank you very much.

In the function "local_align_loss" of the code "vitaa/models/embeddings/seg_head/loss.py", should this line "forward_k_idx = rank1[i, :topK]" be modified as "forward_k_idx = rank1[j, :topK]"?

请问CUHK-PEDES数据集怎么下载的？发邮件也没用呀？

Hello, thanks for your great work! I just want to know the details about HRNet implementation, beacause I can't find anything to do with HRNet in your codebase.

Looking forward your reply, thank your very lot.

Thanks for sharing the code, I wonder how to generate text_attribute_graph? You didn't mention much about this in your paper. Could you please share the method? Because I want to transfer your method to other datasets for experiment, but I don't know how to generate this file. Thanks again.

I tried running the code at https://github.com/Jarr0d/Human-Parsing-Network to create attribute segmentation on the CUHK-PEDES dataset through the Human Parsing Network, but it could not be done because the format of the dataset was different.
Has anyone solved this problem?

../aten/src/ATen/native/cuda/Loss.cu:271: nll_loss_forward_reduce_cuda_kernel_2d: block: [0,0,0], thread: [0,0,0] Assertion t >= 0 && t < n_classes failed.

Hi, thanks for your work!

I found these lines maybe redundant, which leads to an error:

What' more, it would be great if you can provide a requirements.txt. It took me a little while to chose the version of PyTorch and skimage.