Commit 4bb36073 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 434564713
parent 2bfaeb03
...@@ -43,6 +43,7 @@ ctpu up --name <tpu-name> --zone <zone> --tpu-size=v3-32 --tf-version nightly ...@@ -43,6 +43,7 @@ ctpu up --name <tpu-name> --zone <zone> --tpu-size=v3-32 --tf-version nightly
This model requires TF version `>= 2.5`. Currently, that is only available via a This model requires TF version `>= 2.5`. Currently, that is only available via a
`nightly` build on Cloud. `nightly` build on Cloud.
### Install requirements ### Install requirements
SSH into the TPU host with `gcloud compute ssh <tpu-name>` and execute the SSH into the TPU host with `gcloud compute ssh <tpu-name>` and execute the
...@@ -60,7 +61,7 @@ The configurations can be found in the `configs/experiments` directory. You can ...@@ -60,7 +61,7 @@ The configurations can be found in the `configs/experiments` directory. You can
launch a training job by executing. launch a training job by executing.
```shell ```shell
$ export CONFIG=./official/vision/beta/projects/deepmac_maskrcnn/configs/experiments/deep_mask_head_rcnn_voc_r50.yaml $ export CONFIG=./official/projects/deepmac_maskrcnn/configs/experiments/deep_mask_head_rcnn_voc_r50.yaml
$ export MODEL_DIR="gs://<path-for-checkpoints>" $ export MODEL_DIR="gs://<path-for-checkpoints>"
$ export ANNOTAION_FILE="gs://<path-to-coco-annotation-json>" $ export ANNOTAION_FILE="gs://<path-to-coco-annotation-json>"
$ export TRAIN_DATA="gs://<path-to-train-data>" $ export TRAIN_DATA="gs://<path-to-train-data>"
...@@ -71,7 +72,7 @@ task.train_data.input_path=${TRAIN_DATA},\ ...@@ -71,7 +72,7 @@ task.train_data.input_path=${TRAIN_DATA},\
task.annotation_file=${ANNOTAION_FILE},\ task.annotation_file=${ANNOTAION_FILE},\
runtime.distribution_strategy=tpu" runtime.distribution_strategy=tpu"
$ python3 -m official.vision.beta.projects.deepmac_maskrcnn.train \ $ python3 -m official.projects.deepmac_maskrcnn.train \
--logtostderr \ --logtostderr \
--mode=train_and_eval \ --mode=train_and_eval \
--experiment=deep_mask_head_rcnn_resnetfpn_coco \ --experiment=deep_mask_head_rcnn_resnetfpn_coco \
......
...@@ -15,4 +15,4 @@ ...@@ -15,4 +15,4 @@
"""Imports to configure Mask R-CNN with deep mask heads.""" """Imports to configure Mask R-CNN with deep mask heads."""
# pylint: disable=unused-import # pylint: disable=unused-import
from official.vision.beta.projects.deepmac_maskrcnn.tasks import deep_mask_head_rcnn from official.projects.deepmac_maskrcnn.tasks import deep_mask_head_rcnn
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.deepmac_maskrcnn.configs import deep_mask_head_rcnn from official.projects.deepmac_maskrcnn.configs import deep_mask_head_rcnn
class DeepMaskHeadRcnnConfigTest(tf.test.TestCase): class DeepMaskHeadRcnnConfigTest(tf.test.TestCase):
......
...@@ -20,7 +20,7 @@ from absl import logging ...@@ -20,7 +20,7 @@ from absl import logging
import tensorflow as tf import tensorflow as tf
from official.modeling import tf_utils from official.modeling import tf_utils
from official.vision.beta.projects.deepmac_maskrcnn.modeling.heads import hourglass_network from official.projects.deepmac_maskrcnn.modeling.heads import hourglass_network
class DeepMaskHead(tf.keras.layers.Layer): class DeepMaskHead(tf.keras.layers.Layer):
......
...@@ -20,7 +20,7 @@ from absl.testing import parameterized ...@@ -20,7 +20,7 @@ from absl.testing import parameterized
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.deepmac_maskrcnn.modeling.heads import instance_heads as deep_instance_heads from official.projects.deepmac_maskrcnn.modeling.heads import instance_heads as deep_instance_heads
class MaskHeadTest(parameterized.TestCase, tf.test.TestCase): class MaskHeadTest(parameterized.TestCase, tf.test.TestCase):
......
...@@ -21,8 +21,8 @@ from absl.testing import parameterized ...@@ -21,8 +21,8 @@ from absl.testing import parameterized
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.deepmac_maskrcnn.modeling import maskrcnn_model from official.projects.deepmac_maskrcnn.modeling import maskrcnn_model
from official.vision.beta.projects.deepmac_maskrcnn.modeling.heads import instance_heads as deep_instance_heads from official.projects.deepmac_maskrcnn.modeling.heads import instance_heads as deep_instance_heads
from official.vision.modeling.backbones import resnet from official.vision.modeling.backbones import resnet
from official.vision.modeling.decoders import fpn from official.vision.modeling.decoders import fpn
from official.vision.modeling.heads import dense_prediction_heads from official.vision.modeling.heads import dense_prediction_heads
......
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