aap9002/RGB_Optic_Flow_Bend_Classification
Viewer • Updated • 26.2k • 39
How to use aap9002/RGB_Optic_Flow_Bend_Classification with Keras:
# !pip install -U keras tensorflow huggingface_hub
# Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here;
# "jax" and "torch" also work for computation once TensorFlow is installed.
import os
os.environ["KERAS_BACKEND"] = "tensorflow"
import keras
model = keras.saving.load_model("hf://aap9002/RGB_Optic_Flow_Bend_Classification")
Bend Classification Models
This repository organises our trained models for classifying bend sharpness using time-sequence data from two datasets: RGB and Wide View Dense Optic Flow. Overview
Model Varients:
Our dataset:
https://huggingface.co/datasets/aap9002/RGB_Optic_Flow_Bend_Classification
Our Training Script:
https://github.com/AAP9002/Third-Year-Project/blob/main/nn/left_right_classification.ipynb
# !pip install -U keras tensorflow huggingface_hub # Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here; # "jax" and "torch" also work for computation once TensorFlow is installed. import os os.environ["KERAS_BACKEND"] = "tensorflow" import keras model = keras.saving.load_model("hf://aap9002/RGB_Optic_Flow_Bend_Classification")