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 |