def ancMain():
net = YoloNet.AnchorYoloNet()
criterion = Loss.AnchorLoss()
if input('Do you want to load network?').upper() == 'N':
optimizer = optim.Adam(net.parameters(), lr=1e-4)
train_data = Dataset.DetectionDataSet(paths='anc_numpy/paths.txt',
label_dir='anc_numpy/',
root_dir='train/')
trainer = Trainer.Trainer(net=net,
data_set=train_data,
optimizer=optimizer,
criterion=criterion,
save='anc.pt')
trainer.train()
else:
net.load_state_dict(torch.load('anc_cpu.pt'))
test_data = Dataset.DetectionDataSet(paths='anc_numpy_test/paths.txt',
label_dir='anc_numpy_test/',
root_dir='test/')
tester = Tester.AnchorTester(net=net,
test_criterion=criterion,
data_set=test_data)
tester.test()
