dragon-notdragon

Dragon detector with Tensor Flow

This is a simple tensorflow model to detect dragon in images. If you just want to test the trained model, make sure you have the following packages:

tensorflow keras sklearn-deap datasets transformers[torch] sentencepiece

Predict

To run prediction you need to run below code:

from huggingface_hub import from_pretrained_keras

model = from_pretrained_keras("hadilq/dragon-notdragon")

img = keras.preprocessing.image.load_img(filename, target_size=(224, 224))
x = keras.preprocessing.image.img_to_array(img)
x = np.expand_dims(x, axis=0)
x = keras.applications.vgg16.preprocess_input(x)
prediction = model.predict(x)
print("model:", filename, "dragon" if prediction[0][0] >= 0.99 else "notdragon")

Additionally, you can check https://replicate.com/hadilq/dragon-notdragon to play around.

Training procedure

I trained it in Google colab, where you can find the original code in training directory.

Training hyperparameters

The following hyperparameters were used during training:

Model Plot

<details> <summary>View Model Plot</summary>

</details>