Complex-YOLOv4-Pytorch
Complex-YOLOv4-Pytorch copied to clipboard
maudzung
The PyTorch Implementation based on YOLOv4 of the paper: "Complex-YOLO: Real-time 3D Object Detection on Point Clouds"
## What The current repo doesn't support Random Translation augmentation methods. This PR adds Random Translation methods here: https://github.com/maudzung/Complex-YOLOv4-Pytorch/blob/ffb67cd38ee72fd0ad884bcb2f9b11b0cc131f54/src/data_process/transformation.py#L376-L403 ## Why It's reasonable since self-driving car requires a fixed view...
test.py
 When I run the test. py file, there is no result output, and the code does not stop? What is the reason?
@maudzung Hi，Maudzung, nice work! I'm trying to use TensorRT to speed up the inference. Do you have some scripts that transfer the Complex-YOLOv4-Pytorch from pytorch(pth file) to TensorRT? Thanks!
can someone please finish the command of the train.py file, because there is a train.py --gpu_idx 0 --batch_size --num_workers ... and i don't know what is the rest of that...
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 3.79 GiB total capacity; 2.48 GiB already allocated; 25.50 MiB free; 2.52 GiB reserved in total by PyTorch)...
A brief overview of changes done and the files affected mentioned below. - Retaining box prediction confidence when converting the box from LiDAR coordinate frame to camera coordinate frame. File...
https://github.com/maudzung/Complex-YOLOv4-Pytorch/blob/564e8e35ad81f5a9f1a24ca4ceaf10908a100bfd/src/utils/cal_intersection_rotated_boxes.py#L78 I think there should be limitations to the range of intersection points. In the following case, intersection values are calculated as 400.0 even though the boxes do not intersect....
I trained the model with only pedestrians. I tested it and noticed a lot of false positives, I didn't train for too long so I was expecting this. When I...
