cosense3d.agents.cav_prototype package

Submodules

cosense3d.agents.cav_prototype.base_cav module

class cosense3d.agents.cav_prototype.base_cav.BaseCAV(id: str, mapped_id: int, is_ego: bool, lidar_range: Tensor, memory_len: int, lidar_pose: Tensor | None = None, require_grad: bool = False, seq_len: int = 1, **kwargs)

Bases: object

apply_transform()

forward(tasks, training_mode, **kwargs)

forward_fusion(tasks, training_mode, **kwargs)

To be overloaded.

forward_head(tasks, training_mode, **kwargs)

To be overloaded.

forward_local(tasks, training_mode, **kwargs)

To be overloaded.

forward_localization(tasks, training_mode, **kwargs)

To be overloaded.

get_request_cpm()

get_response_cpm()

has_request()

loss(tasks, **kwargs)

To be overloaded.

post_update_memory()

Update memory after each forward run of a single frame.

pre_update_memory()

Update memory before each forward run of a single frame.

prepare_data()

receive_request(request)

receive_response(response)

reset_data(*args, **kwargs)

transform_data()

update(lidar_pose, is_ego, require_grad)

class cosense3d.agents.cav_prototype.base_cav.BaseSeqCAV(id, mapped_id, is_ego, lidar_range, memory_len, lidar_pose=None, require_grad=False, seq_len=1, **kwargs)

Bases: object

apply_transform(seq_idx)

forward(tasks, training_mode, seq_idx, with_loss)

forward_fusion(tasks, training_mode, seq_idx, with_loss)

To be overloaded.

forward_head(tasks, training_mode, seq_idx, with_loss)

To be overloaded.

forward_local(tasks, training_mode, seq_idx, with_loss)

To be overloaded.

get_data(keys, seq_idx=None)

get_request_cpm()

get_response_cpm()

has_request()

loss(tasks, training_mode, seq_idx, with_loss)

To be overloaded.

post_update_memory(seq_idx, **kwargs)

Update memory after each forward run of a single frame.

pre_update_memory(seq_idx, **kwargs)

Update memory before each forward run of a single frame.

prepare_data(seq_idx)

receive_request(request)

receive_response(response, seq_idx)

reset_data(*args, **kwargs)

task_id(seq_idx)

update(lidar_pose)

class cosense3d.agents.cav_prototype.base_cav.DairV2XCAV(*args, **kwargs)

Bases: BaseCAV

class cosense3d.agents.cav_prototype.base_cav.OPV2VtCAV(*args, **kwargs)

Bases: BaseCAV

class cosense3d.agents.cav_prototype.base_cav.OPV2VtCAV_v2(*args, **kwargs)

Bases: BaseCAV

cosense3d.agents.cav_prototype.streamLTS_collection module

class cosense3d.agents.cav_prototype.streamLTS_collection.LTSCAVLocCorr(*args, **kwargs)

Bases: StreamLidarCAV

apply_transform()

forward_fusion(tasks, training_mode, **kwargs)

To be overloaded.

forward_head(tasks, training_mode, **kwargs)

To be overloaded.

forward_local(tasks, training_mode, **kwargs)

To be overloaded.

forward_localization(tasks, training_mode, **kwargs)

To be overloaded.

get_response_cpm()

prepare_data()

transform_data()

class cosense3d.agents.cav_prototype.streamLTS_collection.LTSDairV2X(*args, **kwargs)

Bases: StreamLidarCAV

forward_fusion(tasks, training_mode, **kwargs)

To be overloaded.

forward_head(tasks, training_mode, **kwargs)

To be overloaded.

forward_local(tasks, training_mode, **kwargs)

To be overloaded.

loss(tasks, **kwargs)

To be overloaded.

class cosense3d.agents.cav_prototype.streamLTS_collection.StreamLidarCAV(*args, **kwargs)

Bases: BaseCAV

apply_transform()

forward_fusion(tasks, training_mode, **kwargs)

To be overloaded.

forward_head(tasks, training_mode, **kwargs)

To be overloaded.

forward_local(tasks, training_mode, **kwargs)

To be overloaded.

get_response_cpm()

loss(tasks, **kwargs)

To be overloaded.

post_update_memory()

Update memory after each forward run of a single frame.

pre_update_memory()

Update memory before each forward run of a single frame.

prepare_data()

prepare_time_scale()

refresh_memory(prev_exists)

property timestamp

transform_data()

transform_ref_pts(reference_points, matrix)

update_memory_timestamps(ref_pts)

vis_local_detection()

vis_local_pred()

vis_poses(ax=None, label=None, his_len=1, **kwargs)

vis_ref_pts(ax=None, label=None, his_len=1, **kwargs)

cosense3d.agents.cav_prototype.streamLTS_collection.slcAttnFusion

alias of slcDenseToSparse

class cosense3d.agents.cav_prototype.streamLTS_collection.slcCIASSD(*args, **kwargs)

Bases: StreamLidarCAV

apply_transform()

forward_fusion(tasks, training_mode, **kwargs)

To be overloaded.

forward_head(tasks, training_mode, **kwargs)

To be overloaded.

forward_local(tasks, training_mode, **kwargs)

To be overloaded.

get_response_cpm()

loss(tasks, **kwargs)

To be overloaded.

post_update_memory()

Update memory after each forward run of a single frame.

pre_update_memory()

Update memory before each forward run of a single frame.

prepare_data()

class cosense3d.agents.cav_prototype.streamLTS_collection.slcDenseToSparse(*args, **kwargs)

Bases: StreamLidarCAV

forward_local(tasks, training_mode, **kwargs)

To be overloaded.

prepare_data()

class cosense3d.agents.cav_prototype.streamLTS_collection.slcFPVRCNN(*args, **kwargs)

Bases: StreamLidarCAV

forward_local(tasks, training_mode, **kwargs)

To be overloaded.

prepare_data()

cosense3d.agents.cav_prototype.streamLTS_collection.slcFcooper

alias of slcDenseToSparse

class cosense3d.agents.cav_prototype.streamLTS_collection.slcNoBoxTime(*args, **kwargs)

Bases: StreamLidarCAV

prepare_data()

update_memory_timestamps(ref_pts)

class cosense3d.agents.cav_prototype.streamLTS_collection.slcNoBoxTimeDairV2X(*args, **kwargs)

Bases: LTSDairV2X

prepare_data()

update_memory_timestamps(ref_pts)

Module contents

cosense3d.agents.cav_prototype.get_prototype(module_full_path: str)