Closed s1mpleee closed 2 years ago
this demo file is no longer supported. You can see the instructions in the earlier branch if you want to use it https://github.com/tianweiy/CenterPoint/tree/607f01a46a447d83c12d56eb9f699d69422ebd0f
it gets a link to the demo folder.
this demo file is no longer supported. You can see the instructions in the earlier branch if you want to use it https://github.com/tianweiy/CenterPoint/tree/607f01a46a447d83c12d56eb9f699d69422ebd0f
it gets a link to the demo folder.
thanks for your reply, in fact, I really wonder how can I test the model speed during inference, which file should I run to get the fps result.
this is a bit complicated and most previous reported numbers are a bit noisy. The less rigorous one (the one we do in the original paper) is just using dist_test.py with the flag "--speed_test" and a batch size of 1. But this result is not that accurate, as we hide the latency of voxelization in the dataloader (through multiprocessing)
To be more rigorous, the way Waymo benchmark the latency is through a script like this https://github.com/tianweiy/CenterPoint/blob/1acc72ac7f1e9e21d0c38a2c4353e8b97f343336/tools/simple_inference_waymo.py#L139
You will add a function test_time
import time
def test_time(func):
def inner(*args, **kwargs):
torch.cuda.synchronize()
tic = time.perf_counter()
data_dict = func(*args, **kwargs)
torch.cuda.synchronize()
print(time.perf_counter()-tic)
return data_dict
return inner
and do
test_time(process_example)(...)
In this way, we didn't count the IO time but all the other latency should be accurate
accurate
thanks a lot, another issue occurs to me is that when I run the single_inference.py, I couldn't install these python libary needed for running it.
import rospy
import ros_numpy
and
from std_msgs.msg import Header
import sensor_msgs.point_cloud2 as pc2
from sensor_msgs.msg import PointCloud2, PointField
from jsk_recognitio:_msgs.msg import BoundingBox, BoundingBoxArray
Is there a install_guide or something can help me through it?
Hi, actually, not really sure if you want to use this file. It is for inference with ros. The dist_test.py is the default file for inference using standard pytorch.
tried to replace the demo_infos.pkl with anno file under:
but got an error like this
I assumed that infos_val_10sweeps_withvelo_filter_True.pkl was too big to use in demo.py.