jarr0d / vitaa Goto Github PK

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

Thanks in adavance

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.

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 guess to_bgr255 is used for Caffe2. This should be set to False.

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?

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 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?

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]"?