Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
OpenPCDet
Commits
0c85f610
Commit
0c85f610
authored
Jul 03, 2022
by
Shaoshuai Shi
Browse files
conduct image normalize directly
parent
0743656b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
2 deletions
+1
-2
pcdet/models/backbones_3d/focal_sparse_conv/SemanticSeg/sem_deeplabv3.py
...ckbones_3d/focal_sparse_conv/SemanticSeg/sem_deeplabv3.py
+1
-2
No files found.
pcdet/models/backbones_3d/focal_sparse_conv/SemanticSeg/sem_deeplabv3.py
View file @
0c85f610
...
...
@@ -6,7 +6,6 @@ import torch
import
torch.nn
as
nn
import
torch.nn.functional
as
F
import
torchvision
from
kornia.enhance.normalize
import
normalize
class
SegTemplate
(
nn
.
Module
):
...
...
@@ -112,7 +111,7 @@ class SegTemplate(nn.Module):
# Preprocess images
if
self
.
pretrained
:
images
=
normalize
(
images
,
mean
=
self
.
norm_mean
,
std
=
self
.
norm_std
)
images
=
(
images
-
self
.
norm_mean
[
None
,
:,
None
,
None
].
type_as
(
images
))
/
self
.
norm_std
[
None
,
:,
None
,
None
].
type_as
(
images
)
x
=
images
.
cuda
()
# Extract features
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment