Add documentation and README, use Upsampling2D rather than image Resizing layer

This commit is contained in:
Piv
2021-03-24 21:35:25 +10:30
parent ac3ab27ddd
commit ab7da5acd4
3 changed files with 166 additions and 33 deletions

View File

@@ -2,6 +2,7 @@ import fast_depth_functional as fd
if __name__ == '__main__':
fd.fix_windows_gpu()
model = fd.load_model('fast_depth_nyu_v2_224_224_3_e1')
model = fd.mobilenet_nnconv5_no_bn(weights='imagenet')
fd.compile(model)
fd.train(existing_model=model, save_file='../fast-depth-experimental')
fd.evaluate(model)