deepcrayon
/
tinygrab
1
0
Fork 0
Code

optional look at cat

pull/37/head
George Hotz 2020-10-30 09:30:15 -07:00
parent 2c7e75d733
commit 053c319708
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/efficientnet.py
View File

@ -128,6 +128,13 @@ if __name__ == "__main__":
img -= np.array([0.485, 0.456, 0.406]).reshape((1,-1,1,1))
img /= np.array([0.229, 0.224, 0.225]).reshape((1,-1,1,1))
# if you want to look at the cat
"""
import matplotlib.pyplot as plt
plt.imshow(img[0].mean(axis=0))
plt.show()
"""
# category labels
lbls = fetch("https://gist.githubusercontent.com/aaronpolhamus/964a4411c0906315deb9f4a3723aac57/raw/aa66dd9dbf6b56649fa3fab83659b2acbf3cbfd1/map_clsloc.txt")
lbls = dict([(int(x.split(" ")[1]), x.split(" ")[2]) for x in lbls.decode('utf-8').split("\n")])