Skip to content

Instantly share code, notes, and snippets.

@stardust2602
stardust2602 / convnet_testing.py
Created July 14, 2015 03:08
Caffe script to compute accuracy and confusion matrix based on training input ( .txt file )
#!/usr/bin/python
# -*- coding: utf-8 -*-
import sys
import caffe
import numpy as np
import argparse
from collections import defaultdict
TRAIN_DATA_ROOT='/path/to/training/images/'