danishansari

import os, sys, cv2
import numpy as np
from data_manager import DataManager
from model import get_model, get_cust_model

n_classes = 5
shape = (128, 128, 3)

def train(inp_path):
    trn = DataManager(inp_path, shape, 'TRAIN')

danishansari

import tensorflow as tf

import os, sys
import cv2, numpy as np
from random import shuffle

VALID=0
TRAIN=1

class Classifier():