Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2018-09-03 00:44:55 +02:00
parent defae83d77
commit aa77cbea11
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -65,7 +65,7 @@ def main(opt):
# p.requires_grad = False # p.requires_grad = False
# Set optimizer # Set optimizer
# optimizer = torch.optim.SGD(model.parameters(), lr=.001, momentum=.9, weight_decay=0.0005 * 0, nesterov=True) # optimizer = torch.optim.SGD(model.parameters(), lr=.001, momentum=.9, weight_decay=5e-4, nesterov=True)
# optimizer = torch.optim.Adam(filter(lambda p: p.requires_grad, model.parameters())) # optimizer = torch.optim.Adam(filter(lambda p: p.requires_grad, model.parameters()))
optimizer = torch.optim.Adam(model.parameters()) optimizer = torch.optim.Adam(model.parameters())
optimizer.load_state_dict(checkpoint['optimizer']) optimizer.load_state_dict(checkpoint['optimizer'])

4
utils/utils.py
View File

@ -368,10 +368,10 @@ def plotResults():
plt.figure(figsize=(16, 8)) plt.figure(figsize=(16, 8))
s = ['X', 'Y', 'Width', 'Height', 'Objectness', 'Classification', 'Total Loss', 'Precision', 'Recall'] s = ['X', 'Y', 'Width', 'Height', 'Objectness', 'Classification', 'Total Loss', 'Precision', 'Recall']
for f in ('/Users/glennjocher/Downloads/results.txt', for f in ('/Users/glennjocher/Downloads/results.txt',
'/Users/glennjocher/Downloads/resultsBCE2.txt'): '/Users/glennjocher/Downloads/resultsBCE.txt'):
results = np.loadtxt(f, usecols=[2, 3, 4, 5, 6, 7, 8, 9, 10]).T results = np.loadtxt(f, usecols=[2, 3, 4, 5, 6, 7, 8, 9, 10]).T
for i in range(9): for i in range(9):
plt.subplot(2, 5, i + 1) plt.subplot(2, 5, i + 1)
plt.plot(results[i, :], marker='.', label=f) plt.plot(results[i, :], marker='.', label=f)
plt.title(s[i]) plt.title(s[i])
plt.legend(cocococosadfc) plt.legend()