Commit 997ced7b authored by Scott Zhu's avatar Scott Zhu Committed by A. Unique TensorFlower
Browse files

Export the models directly under vision/modeling as tfm.vision.models.

The models package under the modeling folder serves as a placeholder for now, and eventually all the python modules under modeling now should be moved to modeling/models

PiperOrigin-RevId: 472827931
parent 2e948ccb
...@@ -18,3 +18,4 @@ from official.vision.modeling import backbones ...@@ -18,3 +18,4 @@ from official.vision.modeling import backbones
from official.vision.modeling import decoders from official.vision.modeling import decoders
from official.vision.modeling import heads from official.vision.modeling import heads
from official.vision.modeling import layers from official.vision.modeling import layers
from official.vision.modeling import models
# Copyright 2022 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Models under Vision package."""
from official.vision.modeling.classification_model import ClassificationModel
from official.vision.modeling.maskrcnn_model import MaskRCNNModel
from official.vision.modeling.retinanet_model import RetinaNetModel
from official.vision.modeling.segmentation_model import SegmentationModel
from official.vision.modeling.video_classification_model import VideoClassificationModel
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