Giter VIP home page Giter VIP logo

maudzung / centernet3d-pytorch Goto Github PK

View Code? Open in Web Editor NEW
70.0 5.0 13.0 1.14 MB

Unofficial PyTorch implementation of the paper: "CenterNet3D: An Anchor free Object Detector for Autonomous Driving"

Home Page: https://arxiv.org/pdf/2007.07214.pdf

License: GNU General Public License v3.0

Python 99.86% Shell 0.14%
centernet3d centernet lidar object-detection 3d-object-detection voxel voxelnet sparse-convolution spconv

centernet3d-pytorch's Issues

Pretrained Model and Orientation

Hi,
can you provide pretrained model file and there is an issue with the orientation of the model. I have trained it on Kitti Dataset for 200 epochs and the orientation is still wrong Can you tell what can be the reason? I have attached the results as well please go through it.

Thanks and Regards,
Daniel.
Screenshot from 2021-01-27 10-19-24

Screenshot from 2021-01-27 10-19-36

train.py bugs

configs: {'seed': 2020, 'saved_fn': 'centernet3d', 'root_dir': '../', 'arch': 'centernet3d', 'pretrained_path': None, 'hflip_prob': 0.0, 'no_val': False, 'num_samples': None, 'num_workers': 4, 'batch_size': 16, 'print_freq': 50, 'tensorboard_freq': 50, 'checkpoint_freq': 2, 'start_epoch': 1, 'num_epochs': 50, 'lr_type': 'one_cycle', 'lr': 0.00225, 'minimum_lr': 1e-07, 'momentum': 0.949, 'weight_decay': 0.01, 'optimizer_type': 'adam', 'steps': [150, 180], 'world_size': -1, 'rank': -1, 'dist_url': 'tcp://127.0.0.1:29500', 'dist_backend': 'nccl', 'gpu_idx': 0, 'no_cuda': False, 'multiprocessing_distributed': False, 'evaluate': False, 'resume_path': None, 'K': 100, 'device': device(type='cuda', index=0), 'ngpus_per_node': 1, 'pin_memory': True, 'sparse_shape': [40, 1400, 1600], 'input_size': [1400, 1600], 'hm_size': [350, 400], 'down_ratio': 4, 'max_objects': 50, 'head_conv': 64, 'num_classes': 1, 'num_center_offset': 2, 'num_z': 1, 'num_dim': 3, 'num_direction': 2, 'num_conners': 4, 'heads': {'hm_cen': 1, 'cen_offset': 2, 'direction': 2, 'z_coor': 1, 'dim': 3, 'hm_conners': 1}, 'num_input_features': 4, 'dataset_dir': '../dataset/kitti', 'checkpoints_dir': '../checkpoints/centernet3d', 'logs_dir': '../logs/centernet3d', 'distributed': False, 'subdivisions': 4, 'is_master_node': True}
QObject::moveToThread: Current thread (0x5632b7c00070) is not the object's thread (0x5632b83c94d0).
Cannot move to target thread (0x5632b7c00070)

qt.qpa.plugin: Could not load the Qt platform plugin "xcb" in "/home/hy/anaconda3/envs/pointpillars/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.

I met this error but I don't know how to deal it. Please help me

No results following train.sh

Hi,

First of all, thank you for your excellent work. Your code structure is very clean.

However, when I trained with your code with “train.sh”, and after 50 epochs, I found no bboxs and no results with "python test.py".(with peak_thresh=0.2). Could you give me some advice?

By the way, could you provide a pre-trained model so that I can have a further understanding of the output results of the whole network.

Look to your reply and wish you have a nice day!

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.