cvip-labs 0.2.0

CVIP 6th sem lab helpers - MNIST and CNN experiments

cvip_labs

CVIP 6th sem lab helpers.

Install

pip install cvip_labs

Exp 4 - MNIST SVM + Neural Network

from cvip_labs import run_exp4 X_train, X_test, y_train, y_test = run_exp4.load_data() svm_model = run_exp4.train_svm(X_train, y_train) svm_acc = run_exp4.evaluate(svm_model, X_test, y_test) nn_model = run_exp4.train_nn(X_train, y_train, epochs=5) nn_acc = run_exp4.evaluate_nn(nn_model, X_test, y_test) run_exp4.plot_results()

Exp 5 - CNN Image Classifier

from cvip_labs import run_exp5 train, val, test, class_names = run_exp5.load_data(data_dir='exp5') model = run_exp5.build_model() model.summary() hist = run_exp5.train_model(model, train, val, epochs=20) run_exp5.plot_results(hist) run_exp5.evaluate(model, test) run_exp5.predict(model, class_names, img_path='images6.jpg') run_exp5.save_model(model)