Easy-to-use VGGNet
- Prepare your Dataset.
- Modify
setting.yamlfile. (Check out the# write herecomments section.) - Run
python main.py.
-
model.fit(x, y, validation_data, epochs=30, batch_size=4)
: Training function is reconstructed like tensorflow-keras style. -
model.evaluate(model, dataloader, valid_iter=1, batch_size=1, criterion=None) -
model.predict(test_images, use_cpu=False)