Git Product home page Git Product logo

lizhaoliu-lec / cpcm Goto Github PK

View Code? Open in Web Editor NEW
28.0 3.0 3.0 6 MB

This is the official repo for Contextual Point Cloud Modeling for Weakly-supervised Point Cloud Semantic Segmentation (ICCV 23).

License: MIT License

Python 95.84% Shell 4.16%
3d-segmentation augmentation consistency-regularization context-model data-efficient-learning point-cloud pytorch-implementation segmentation semi-supervised-learning sparse-annotations

cpcm's People

Contributors

lizhaoliu-lec avatar xiaoxunlong avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

cpcm's Issues

Question about the training script for the 0.02% (S3DIS)

Hi.
Thank you for releasing the code.
However, I cannot find the training script for the 0.02% setting on the S3DIS dataset.
I modified the script for 0.01% as 0.02% and tried it.
Unfortunately, this setting fails to reproduce the reported performance.
Could you provide the official training script for it?
Any helpful comment will be greatly appreciated.

merge_sort: failed to synchronize: cudaErrorIllegalAddress: an illegal memory access was encountered terminate called after throwing an instance of 'c10::Error'

Thanks for your code! I have encountered an error when training:

Traceback (most recent call last):
File "ddp_train.py", line 116, in
main()
File "ddp_train.py", line 107, in main
trainer.train()
File "/home/Point_Cloud/CPCM/trainer/base.py", line 190, in train
self.train_one_epoch()
File "/home/Point_Cloud/CPCM/trainer/fully_supervised_trainer.py", line 332, in train_one_epoch
step_ret = self.step(batch)
File "/home/Point_Cloud/CPCM/trainer/fully_supervised_trainer.py", line 1232, in step
return self._step_two_and_mask_stream(batch=batch)
File "/home/Point_Cloud/CPCM/trainer/fully_supervised_trainer.py", line 1199, in _step_two_and_mask_stream
loss.backward()
File "/home/.conda/envs/seg/lib/python3.8/site-packages/torch/_tensor.py", line 307, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph, inputs=inputs)
File "/home/.conda/envs/seg/lib/python3.8/site-packages/torch/autograd/init.py", line 154, in backward
Variable._execution_engine.run_backward(
RuntimeError: merge_sort: failed to synchronize: cudaErrorIllegalAddress: an illegal memory access was encountered
terminate called after throwing an instance of 'c10::Error'

How can I fix this? I tried this taesungp/contrastive-unpaired-translation#83 but did not success.

Cannot reproduce the results.

Thank you for your codes. I have found that the setting of two_stream_mask_self_loss in default.yaml is "False", which means that the calculation of mask loss for feats_aux and feats_masked_aux (Z2, Zm loss in your paper) is not involved. I also found that two_stream_mask_mode was set to "random", while it should be "grid". I have corrected them and test under 0.1% setting on S3DIS, while the mIoU is only 63.5, not 66.3.

Question about the requirement of GPU

Hi! I wonder how many GPUs and what kind of GPUs are required in general weakly-supervised semantic segmentation task including your work? Plus as introduced in your paper, 1 or 2 TITAN 3090 GPU is enough, so how long does it take for a single group of experiment? Thanks!

How to test?

Thank you for such a good job. I have a small question. Trained, how to test. Which parameters should be changed?

question about multi-GPUs training

thank you for your excellent work.
because my sigle gpu's memory isn't enough, I want to know if your code can be modified use multi-gpus to train.

Cannot reproduce the results

Thanks for the great research, I'm having some issues running this. MinkowskiEngine 0.5.4 is used. During the process of preparing S3DIS data, an error occurred and the return value was incorrect. However, it can run normally after subsequent modifications. However, when running CPCM's 0.1S3DIS data alone, val_miou is 56%, and both the script and default.yaml are the latest in the project. Looking forward to your reply。
image

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.