No public description

PiperOrigin-RevId: 858967668

Author: tensorflower-gardener

official/projects/waste_identification_ml/Deploy/detr_cloud_deployment/client/requirements.sh

@@ -0,0 +1,48 @@
+#!/bin/bash
+
+# Summary
+cat << EOF
+This script sets up the environment for running ML models by ensuring Bash
+execution, installing system dependencies, setting up a virtual environment,
+installing ML packages, and cloning TensorFlow Model Garden.
+EOF
+
+# Ensure the script is executed with /bin/bash
+if [ -z "$BASH_VERSION" ]; then
+  exec /bin/bash "$0" "$@"
+fi
+
+sudo apt-get update -y
+
+# Install Docker if not already installed
+if ! command -v docker &> /dev/null
+then
+  echo "Docker is not installed. Installing Docker..."
+  curl -fsSL https://get.docker.com -o get-docker.sh
+  sudo sh get-docker.sh
+  rm -f get-docker.sh
+  echo "Docker installation completed."
+else
+  echo "Docker is already installed. Skipping Docker installation."
+fi
+
+# Create a virtual environment and install packages
+sudo apt-get install -y python3-venv python3-pip
+
+python3.10 -m venv myenv
+source myenv/bin/activate
+
+pip install --no-cache-dir natsort absl-py opencv-python pandas pandas-gbq \
+  google-cloud-bigquery google-auth trackpy google-cloud-storage \
+  scikit-image scikit-learn webcolors==1.13 ffmpeg-python tritonclient[all] \
+  supervision==0.26.1 pillow==12.0.0
+
+# Clone TensorFlow Model Garden if the 'models' directory does not exist
+if [ ! -d "models" ]; then
+  git clone --depth 1 https://github.com/tensorflow/models.git
+else
+  echo "'models' directory already exists. Skipping cloning."
+fi
+
+deactivate
+echo "Environment setup is complete."

official/projects/waste_identification_ml/Deploy/detr_cloud_deployment/client/run_images.sh

@@ -0,0 +1,58 @@
+#!/bin/bash
+
+cat << EOF
+This script automates the execution of an Circularnet pipeline for image
+processing.
+Steps Performed:
+ 1. Activates the Python virtual environment named 'myenv'.
+ 2. Validates successful activation of the virtual environment.
+ 3. Executes the 'pipeline_images.py' script with the following parameters:
+ Parameters:
+ --input_directory         : GCS directory where the input images are stored for
+                             inference.
+ --output_directory        : GCS directory where the model inference outputs will be
+                             saved.
+ --height                  : Height to which input images are resized for the Mask
+                             R-CNN model.
+ --width                   : Width to which input images are resized for the Mask
+                             R-CNN model.
+ --model                   : Name of the model to download and use for inference.
+ --score                   : Confidence threshold for detections during
+                             inference.
+ --search_range_x          : Max pixel movement allowed in the X direction for
+                             object tracking between missed frames.
+ --search_range_y          : Max pixel movement allowed in the Y direction for
+                             object tracking between missed frames.
+  --memory                 : Number of frames an object can be missed and still
+                             be tracked.
+  --project_id             : Google Cloud Project ID for BigQuery operations.
+  --bq_dataset_id          : BigQuery Dataset ID where results will be stored.
+  --bq_table_id            : BigQuery Table ID where results will be stored.
+  --overwrite              : If set to True, overwrites the pre-existing
+                             BigQuery table.
+  --tracking_visualization : If set to True, visualizes the tracking results
+                             from the tracking algorithm.
+  --cropped_objects        : If set to True, crops the objects per category
+                             according to the prediction and tracking results.
+EOF
+#Activate the virtual environment
+source myenv/bin/activate
+# Check if the virtual environment is activated
+if [[ "$VIRTUAL_ENV" != "" ]]; then
+    echo "Virtual environment 'myenv' activated successfully."
+else
+    echo "Failed to activate virtual environment. Exiting."
+    exit 1
+fi
+python inference_pipeline.py \
+	--input_directory=gs://recykal/TestData/SmallTestData \
+	--output_directory=gs://recykal/TestData/SmallTestData \
+	--model=detr_seg \
+	--score=0.50 \
+	--search_range_x=150 \
+	--search_range_y=20 \
+	--memory=3  \
+	--project_id=waste-identification-ml-330916 \
+	--bq_dataset_id=circularnet_dataset \
+	--bq_table_id=vinit_test_table1 \
+	--overwrite=True