Skip to content

Commit

Permalink
remove deeplabv3 dualhead warning and add link for deeplabv3_dualhead.py
Browse files Browse the repository at this point in the history
  • Loading branch information
remtav committed Mar 24, 2022
1 parent bd01d7d commit bd0c88a
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 9 deletions.
2 changes: 2 additions & 0 deletions models/deeplabv3_dualhead.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# See: https://www.azavea.com/blog/2019/08/30/transfer-learning-from-rgb-to-multi-band-imagery/

import logging
from typing import Optional

Expand Down
9 changes: 0 additions & 9 deletions train_segmentation.py
Original file line number Diff line number Diff line change
Expand Up @@ -780,15 +780,6 @@ def main(cfg: DictConfig) -> None:
-------
:param cfg: (dict) Parameters found in the yaml config file.
"""
# Limit of the NIR implementation
# FIXME: keep this warning?
# if 'deeplabv3' not in cfg.model.model_name and 'IR' in read_modalities(cfg.dataset.modalities):
# logging.info(
# '\nThe NIR modality will be fed at first layer of model alongside other bands,'
# '\nthe implementation of concatenation point at an intermediary layer is only available'
# '\nfor the deeplabv3 model for now. \nMore will follow on demand.'
# )

# Preprocessing
# HERE the code to do for the preprocessing for the segmentation

Expand Down

0 comments on commit bd0c88a

Please sign in to comment.