Commit 0b882582 authored by liyinhao's avatar liyinhao
Browse files

change to fstring

parent 36466f83
# Deep Hough Voting for 3D Object Detection in Point Clouds
## Introduction
```
@inproceedings{qi2019deep,
author = {Qi, Charles R and Litany, Or and He, Kaiming and Guibas, Leonidas J},
title = {Deep Hough Voting for 3D Object Detection in Point Clouds},
booktitle = {Proceedings of the IEEE International Conference on Computer Vision},
year = {2019}
}
```
## Results and models
| Backbone | Style | Lr schd | Mem (GB) | Inf time (fps) | box AP | Download |
## Different regression loss
| Backbone | Loss type | Mem (GB) | Inf time (fps) | box AP | Download |
## Pre-trained Models
| Backbone | Style | Lr schd | Mem (GB) | Inf time (fps) | box AP | Download |
...@@ -47,7 +47,7 @@ def export_one_scan(scan_name, output_filename_prefix, max_num_point, ...@@ -47,7 +47,7 @@ def export_one_scan(scan_name, output_filename_prefix, max_num_point,
bbox_mask = np.in1d(instance_bboxes[:, -1], OBJ_CLASS_IDS) bbox_mask = np.in1d(instance_bboxes[:, -1], OBJ_CLASS_IDS)
instance_bboxes = instance_bboxes[bbox_mask, :] instance_bboxes = instance_bboxes[bbox_mask, :]
print('Num of care instances: ', instance_bboxes.shape[0]) print(f'Num of care instances: {instance_bboxes.shape[0]}')
N = mesh_vertices.shape[0] N = mesh_vertices.shape[0]
if N > max_num_point: if N > max_num_point:
...@@ -82,7 +82,7 @@ def batch_export(max_num_point, output_folder, train_scan_names_file, ...@@ -82,7 +82,7 @@ def batch_export(max_num_point, output_folder, train_scan_names_file,
export_one_scan(scan_name, output_filename_prefix, max_num_point, export_one_scan(scan_name, output_filename_prefix, max_num_point,
label_map_file, scannet_dir) label_map_file, scannet_dir)
except Exception: except Exception:
print('Failed export scan: %s' % (scan_name)) print(f'Failed export scan: {scan_name}')
print('-' * 20 + 'done') print('-' * 20 + 'done')
......
...@@ -72,7 +72,7 @@ def export(mesh_file, ...@@ -72,7 +72,7 @@ def export(mesh_file,
output_file(str): Path of the output folder. output_file(str): Path of the output folder.
Default: None. Default: None.
Returns: It returns a tuple, which containts the the following things:
ndarray: Vertices of points data. ndarray: Vertices of points data.
ndarray: Indexes of label. ndarray: Indexes of label.
ndarray: Indexes of instance. ndarray: Indexes of instance.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment