Skip to content

Instantly share code, notes, and snippets.

View metal3d's full-sized avatar
Electrifying my brain

Patrice Ferlet metal3d

Electrifying my brain
View GitHub Profile
@metal3d
metal3d / train.py
Created June 28, 2021 12:16
Script to train a binary model to help on creating dataset
""" Train model and find bad image from a directory """
import argparse
import os
import time
from glob import glob
from typing import Any
import numpy as np
from tensorflow.python.keras.layers.pooling import MaxPool2D
@metal3d
metal3d / Info
Created June 16, 2021 06:47
Countries coordinates with longitude and latitude
Based on https://gist.github.com/tadast/8827699:
- remove duplicates
- striped and no unecessary double quotes
- floating point values instead of strings
<html>
<meta charset="utf-8" />
<body>
<pre>
<code>
# All you have to do is to prefix variables with "_change_"
# and let the script to manage the editable values
FOO="_change_thisvariable"
this is a command with _change_argument here
@metal3d
metal3d / sun_set.py
Last active October 14, 2020 07:35
Align Sky textyure with the sun lamp in Blender
import bpy
import mathutils
# Author: Patrice Ferlet <metal3d@gmail.com>
# License: GPL
# Works with Blender 2.80-2.90
# To use this script:
# - paste this script in a "text editor" *inside* blender
# - name it "sun_set.py"
@metal3d
metal3d / videoframegenerator.py
Last active August 7, 2023 10:24
A video frame generator respecting Keras.Sequence class with data augmentation capacities
# DEPRECATED - please go to my complete package:
# pipy package https://pypi.org/project/keras-video-generators/
# github sources https://github.com/metal3d/keras-video-generators
import keras
import cv2 as cv
import glob
import numpy as np
import os
import random
@metal3d
metal3d / videoframegenerator.py
Created April 2, 2019 12:16
A video frame generator respecting Keras.Sequence class with data augmentation capacities
import keras
import cv2 as cv
import glob
import numpy as np
import os
import random
import keras_preprocessing
# author: Patrice Ferlet <patrice.ferlet@smile.fr>
# licence: MIT
@metal3d
metal3d / TF - graph exemple.ipynb
Last active January 20, 2019 16:37
Tensorflow basics
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
import tensorflow as tf
import numpy as np
from IPython.display import clear_output, Image, display, HTML
def strip_consts(graph_def, max_const_size=32):
"""Strip large constant values from graph_def."""
strip_def = tf.GraphDef()
for n0 in graph_def.node:
n = strip_def.node.add()
n.MergeFrom(n0)
@metal3d
metal3d / generator.py
Last active December 7, 2022 22:41
Keras generator to create sequence image batches
import cv2 as cv
import os
import glob
import numpy as np
import random
from tensorflow import keras
import keras.preprocessing.image
from tensorflow.python.keras.utils.data_utils import Sequence
class VideoFrameGenerator(Sequence):
@metal3d
metal3d / make_prediction.py
Last active November 28, 2018 12:33
Keras make prediction on image sequence
from IPython.display import HTML, display
import cv2 as cv
import os
import matplotlib.pyplot as plt
def make_prediction(model, movie, nframe, classes, shape=(224, 224), nchan=3):
video = cv.VideoCapture(movie)
frames = []
while True:
grabbed, frame = video.read()