Skip to content

Commit

Permalink
Updated Tensorflow brains
Browse files Browse the repository at this point in the history
  • Loading branch information
sergiopaniego committed Sep 6, 2023
1 parent ac3e377 commit 43e2309
Show file tree
Hide file tree
Showing 14 changed files with 1 addition and 11 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import carla
from os import path
from albumentations import (
Compose, Normalize, RandomRain, RandomBrightness, RandomShadow, RandomSnow, RandomFog, RandomSunFlare, GridDropout
Compose, Normalize, RandomRain, RandomBrightness, RandomShadow, RandomSnow, RandomFog, RandomSunFlare
)
from utils.constants import PRETRAINED_MODELS_DIR, ROOT_PATH
from utils.logger import logger
Expand Down Expand Up @@ -145,15 +145,6 @@ def execute(self):
bird_eye_view_1
]


AUGMENTATIONS_TEST = Compose([
GridDropout(p=1.0)
])

bird_eye_view_1 = AUGMENTATIONS_TEST(image=bird_eye_view_1)
bird_eye_view_1 = bird_eye_view_1["image"]


self.update_frame('frame_1', image_1)
self.update_frame('frame_2', image_2)
self.update_frame('frame_3', image_3)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,6 @@ def __init__(self, sensors, actuators, handler, model, config=None):
logger.info("File " + model + " cannot be found in " + PRETRAINED_MODELS)
logger.info("** Load TF model **")
self.net = tf.keras.models.load_model(PRETRAINED_MODELS + model)
print(self.net.summary())
logger.info("** Loaded TF model **")
else:
logger.info("** Brain not loaded **")
Expand Down

0 comments on commit 43e2309

Please sign in to comment.