Skip to content

[Feature] Add ceph and petrel support for LoadMultiViewImageFromFiles #1651

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 7 commits into
base: dev-1.0
Choose a base branch
from
Open

[Feature] Add ceph and petrel support for LoadMultiViewImageFromFiles #1651

Changes from 1 commit
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
e0d38a5
[Feature] Add ceph and petrel support for LoadMultiViewImageFromFiles
Lotus-95 Jul 21, 2022
64975b5
fix lint
JingweiZhang12 Sep 29, 2022
c23b9e1
merge from master
JingweiZhang12 Sep 29, 2022
c7569b4
fix link
JingweiZhang12 Sep 29, 2022
8449b99
Bump to 1.0.0rc6 (#2073)
JingweiZhang12 Dec 2, 2022
33a9037
[Doc]: update new releases of 1.1 in readme (#2108)
JingweiZhang12 Dec 9, 2022
47b8f4a
merge from upstream dev
JingweiZhang12 Dec 13, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 6 additions & 2 deletions mmdet3d/datasets/pipelines/loading.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,13 @@ class LoadMultiViewImageFromFiles(object):
Defaults to 'unchanged'.
"""

def __init__(self, to_float32=False, color_type='unchanged'):
def __init__(self,
to_float32=False,
color_type='unchanged',
file_client_args=dict(backend='disk')):
self.to_float32 = to_float32
self.color_type = color_type
self.file_client_args = file_client_args.copy()

def __call__(self, results):
"""Call function to load multi-view image from files.
Expand All @@ -45,7 +49,7 @@ def __call__(self, results):
filename = results['img_filename']
# img is of shape (h, w, c, num_views)
img = np.stack(
[mmcv.imread(name, self.color_type) for name in filename], axis=-1)
[mmcv.imread(name, self.color_type, file_client_args=self.file_client_args) for name in filename], axis=-1)
if self.to_float32:
img = img.astype(np.float32)
results['filename'] = filename
Expand Down