Coder Social home page Coder Social logo

zhangyp15 / occformer Goto Github PK

View Code? Open in Web Editor NEW
306.0 11.0 22.0 60.81 MB

[ICCV 2023] OccFormer: Dual-path Transformer for Vision-based 3D Semantic Occupancy Prediction

Home Page: https://arxiv.org/abs/2304.05316

License: Apache License 2.0

Python 90.73% Shell 0.67% Jupyter Notebook 0.05% Dockerfile 0.02% Makefile 0.03% Batchfile 0.04% CSS 0.01% C++ 5.00% Cuda 3.47%
autonomous-driving multi-camera occupancy-prediction semantic-segmentation semantic-scene-completion semantic-scene-understanding

occformer's People

Contributors

yunpengzhangphigent avatar zhangyp15 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  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  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  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

occformer's Issues

depth

1681996633055

Hello, i run the code of visualization gt_depths with image, why it shows this way. I haven't done the depth estimation before, but i saw the depth images are not shown as this.

furthest_point_sample.py error

Hi, thank you for sharing your work.
However, when I use
python tools/create_data.py nuscenes --root-path ./data/nuscenes --out-dir ./data --version v1.0 --canbus ./data
to generate the info, some thing goes wrong like:
ImportError: /home/guixingtai/anaconda3/envs/occformer/lib/python3.7/site-packages/mmdet3d-0.17.1-py3.7-linux-x86_64.egg/mmdet3d/ops/furthest_point_sample/furthest_point_sample_ext.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _Z39furthest_point_sampling_kernel_launcheriiiPKfPfPiP11CUstream_st

can you help me?
thanks

Efficiency of the method

Congratulations on the amazing work.

I am curious about the efficiency of the method. Could you shed some light on the inference time and the memory requirement?

Clarification on Training Data for the Reported Results on SemanticKITTI Test

Hello,

I would like to seek clarification regarding the training data used for the reported results on the SemanticKITTI test. Specifically, I would like to know whether the results were obtained by training solely on the "train" subset or by combining both the "train" and "val" subsets.

Thanks for your assistance in providing this information.

What's the results without the query-based Transformer decoder?

Congratulations on creating such an excellent and solid work!

However, I'm wondering about the results achieved without the query-based Transformer decoder, or in other words, the isolated impact of the Transformer Occupancy Decoder.
Given that the Dual-path Transformer Encoder guides the Voxel Features with the BEV feature, it seems that the voxel features should already possess sufficient fine-grained features for SSC.
Additionally, the absence of instance-level annotations could possibly reduce the impact of the Transformer Decoder and one-to-one matching.

I would appreciate any insights you may have on this matter.

ModuleNotFoundError: No module named 'visualize_nusc_release'

Hi,
When I run the script python projects/mmdet3d_plugin/visualize/visualize_nusc_video.py $YOUR_SAVE_PATH $YOUR_VISUALIZE_SAVE_PATH,I met the error below.

Traceback (most recent call last):
  File "projects/mmdet3d_plugin/visualize/visualize_nusc_video.py", line 8, in <module>
    from visualize_nusc_release import draw_nusc_occupancy
ModuleNotFoundError: No module named 'visualize_nusc_release'

I cannot find and import the visualize_nusc_release module.

When I try to execute the code after installing PyQT5 using pip install PyQT5 the following error occurs.

QObject::moveToThread: Current thread (0xd4bcc0) is not the object's thread (0x1416720).
Cannot move to target thread (0xd4bcc0)

qt.qpa.plugin: Could not load the Qt platform plugin "xcb" in "/home/limlab/anaconda3/envs/occformer/lib/python3.7/site-packages/cv2/qt/plugins" even though it was found.
This application failed to start because no Qt platform plugin could be initialized. Reinstalling the application may fix this problem.

Available platform plugins are: xcb, eglfs, linuxfb, minimal, minimalegl, offscreen, vnc, wayland-egl, wayland, wayland-xcomposite-egl, wayland-xcomposite-glx, webgl.


Aborted (core dumped)

please help me

“AttributeError: 'NuScenes' object has no attribute 'lidarseg_name2idx_mapping'”

Hi,
When I ran the scrript python projects/mmdet3d_plugin/tools/validate_lidarseg_submission.py --result-path $YOUR_SAVE_PATH --dataroot data/nuscenes --zip-out ., I met the error below.

  File "/rockywin.wang/occ_net/OccFormer/projects/mmdet3d_plugin/tools/validate_lidarseg_submission_my.py", line 158, in <module>
    zip_out=zip_out_)
  File "/rockywin.wang/occ_net/OccFormer/projects/mmdet3d_plugin/tools/validate_lidarseg_submission_my.py", line 35, in validate_submission
    mapper = LidarsegClassMapper(nusc)
  File "/opt/conda/lib/python3.7/site-packages/nuscenes/eval/lidarseg/utils.py", line 140, in __init__
    self.fine_idx_2_coarse_idx_mapping = self.get_fine_idx_2_coarse_idx()
  File "/opt/conda/lib/python3.7/site-packages/nuscenes/eval/lidarseg/utils.py", line 241, in get_fine_idx_2_coarse_idx
    for fine_name, fine_idx in self.nusc.lidarseg_name2idx_mapping.items():
AttributeError: 'NuScenes' object has no attribute 'lidarseg_name2idx_mapping'

The below is my data structure.
image

“KeyError: 'MaskHungarianAssigner is already registered in bbox_assigner'”

Hi,
Thank you for sharing the great work!
When I run the scipt bash tools/dist_test.sh projects/configs/occformer_nusc/occformer_nusc_r50_256x704.py /rockywin.wang/occ_net/OccFormer/occformer_nusc_r50.pth 4 --pred-save /rockywin.wang/occ_net/OccFormer/01_pred_dir
I met the error below.
The version of MMCV is 1.4.0.

te/attempt_3/0/error.json
Traceback (most recent call last):
  File "tools/test.py", line 19, in <module>
    from projects.mmdet3d_plugin.datasets.builder import build_dataloader
  File "/rockywin.wang/occ_net/OccFormer/projects/mmdet3d_plugin/__init__.py", line 1, in <module>
    from .occformer import *
  File "/rockywin.wang/occ_net/OccFormer/projects/mmdet3d_plugin/occformer/__init__.py", line 5, in <module>
    from .mask2former import *
  File "/rockywin.wang/occ_net/OccFormer/projects/mmdet3d_plugin/occformer/mask2former/__init__.py", line 2, in <module>
    from .assigners import *
  File "/rockywin.wang/occ_net/OccFormer/projects/mmdet3d_plugin/occformer/mask2former/assigners/__init__.py", line 1, in <module>
    from .mask_hungarian_assigner import MaskHungarianAssigner
  File "/rockywin.wang/occ_net/OccFormer/projects/mmdet3d_plugin/occformer/mask2former/assigners/mask_hungarian_assigner.py", line 12, in <module>
    class MaskHungarianAssigner(BaseAssigner):
  File "/opt/conda/lib/python3.7/site-packages/mmcv/utils/registry.py", line 312, in _register
    module_class=cls, module_name=name, force=force)
  File "/opt/conda/lib/python3.7/site-packages/mmcv/utils/registry.py", line 246, in _register_module
    raise KeyError(f'{name} is already registered '
KeyError: 'MaskHungarianAssigner is already registered in bbox_assigner'
ERROR:torch.distributed.elastic.multiprocessing.api:failed (exitcode: 1) local_rank: 0 (pid: 172562) of binary: /opt/conda/bin/python
ERROR:torch.distributed.elastic.agent.server.local_elastic_agent:[default] Worker group failed

Expected dtype object, got 'numpy.dtype[uint8]'

When I try to run test.py with one GPU, the terminal reports Expected dtype object, got 'numpy.dtype[uint8] in Line 107 of projects/mmdet3d_plugin/datasets/pipelines/loading_nusc_occ.py. But i have checked Line122-123, nb.jit('u1[:,:,:](u1[:,:,:],i8[:,:])', nopython=True, cache=True, parallel=False) and the datatype matches the processed_label and label_voxel_pair. Can you help me solve this issue? Thx!

fov_mask

i wanna to visualize one photo with mask ,but i don't know how to generate the fov_mask.(the photo is come from the Semantic kitti)
where provide the method among the source code of Occformer?
Please,I have a terrible time on this problem
mask
I need to generate the kind of picture with vision shadow

Can OccFormer have a per-voxel segmentation config instead of mask - cls

Hi there, I was wondering whether the Mask2FormerOccHead can be adapted to use a simple per-voxel loss such as CE (or Geo-Scal loss from MonoScene). I find that using mask and cls paradigm in Mask2Former is kinda difficult to apply known loss function in voxel space. Can you think of a way to adapt this in your code? Thank you

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.