eliphatfs / skeletonmerger Goto Github PK

我想了解一下训练这个对准关键点检测器一般需要显存多少的GPU以及需要训练多长时间？

Hello,

I downloaded the "ShapeNetCore.v2" from https://github.com/AnTao97/PointCloudDatasets, as mentioned in the readme.
I try to train the model using following configurations:
python train.py -t '/home/username/datasets/shapenetcorev2_hdf5_2048' -v '/home/username/datasets/shapenetcorev2_hdf5_2048/val' -c 1 -m airplane-merger.pt -d gpu

And, I get the following error messages:
Traceback (most recent call last):
File "/home/projects/SkeletonMerger/train.py", line 79, in
x, xl = all_h5(DATASET, True, True, subclasses=(ns.subclass,), sample=None) # n x 2048 x 3
File "/home/projects/SkeletonMerger/merger/data_flower.py", line 36, in all_h5
xy = tuple(lazy)
File "/home/projects/SkeletonMerger/merger/data_flower.py", line 33, in
lazy = map(lambda x: load_h5(x, normalize, include_label),
File "/home/projects/SkeletonMerger/merger/data_flower.py", line 14, in load_h5
f = h5py.File(h5_filename, 'r')
File "/home/anaconda3/envs/pytorchworkshop/lib/python3.9/site-packages/h5py/_hl/files.py", line 455, in init
fid = make_fid(name, mode, userblock_size,
File "/home/anaconda3/envs/pytorchworkshop/lib/python3.9/site-packages/h5py/_hl/files.py", line 199, in make_fid
fid = h5f.open(name, flags, fapl=fapl)
File "h5py/_objects.pyx", line 54, in h5py._objects.with_phil.wrapper
File "h5py/_objects.pyx", line 55, in h5py._objects.with_phil.wrapper
File "h5py/h5f.pyx", line 100, in h5py.h5f.open
OSError: Unable to open file (file signature not found)

It seems there is some issue in reading the h5 file. Please let me know.

Hi, nice job

I am very sorry to bother you, when I ran the predictor_keypointnet.py on the guitar category, I met the following error:
99%|█████████▉| 696/704 [03:42<00:02, 3.12it/s]
Traceback (most recent call last):
recon, key_points, kpa, emb, null_activation = net(torch.Tensor(np.array(Q)).to(ns.device))
x = batch_norm(x) merger_net.py", line 35, in forward
ValueError: expected 2D or 3D input (got 1D input)

Thank you very much!

Hi, nice job!

I have noticed that you changed the subclass id of chair category(0 to 14). So , would you please tell me the subclass id of Airplanes and Guitars?

Thank you very much!

Please help on how to train this on kitti

Where can I get the annotation json and pcd file to predict keypoint?
I only have the shapenetcorev2 dataset ,only .h5 files and some json files.
Thanks!

How to use it in shapenetcorev2 data set? How to modify predictor_keypointnet.py and eval_keypointnet.py?
I'm stuck here, please help TAT

I trained your model locally with default settings, however, the result is not as good as shown in the paper. I write this issue to ask for your help. Could you please provide me with your training settings( epoch, batch num, optimizer). Also, I would be very glad to get a pretrained model from you. Best wishes! 😄

Sorry to bother you , i ues shapenetcorev2 dataset, but i don't know how can i divide the dataset just like you? can you help me ?