Switch to unified view

a b/(1) PyTorch_HistoNet/util/getClassCount.py
1
import numpy as np
2
from util import pause
3
4
def getClassCount(classVec):
5
    #print(classVec[0])
6
    classCounts = np.zeros(len(classVec[0]))
7
    for vec in classVec:
8
        classCounts += vec
9
10
        #print(vec)
11
12
    #print('Uscita dalla funzione classcount')
13
    #print(classCounts)
14
    #pause()
15
16
    return classCounts
17