Browse Source

fixed the path combining issue

pull/312/head
Kerry Jiang 6 years ago
parent
commit
0ca0bf909e
2 changed files with 9 additions and 0 deletions
  1. +6
    -0
      src/TensorFlowHub/MnistModelLoader.cs
  2. +3
    -0
      src/TensorFlowHub/Utils.cs

+ 6
- 0
src/TensorFlowHub/MnistModelLoader.cs View File

@@ -93,6 +93,9 @@ namespace Tensorflow.Hub


private NDArray ExtractImages(string file, int? limit = null) private NDArray ExtractImages(string file, int? limit = null)
{ {
if (!Path.IsPathRooted(file))
file = Path.Combine(AppContext.BaseDirectory, file);

using (var bytestream = new FileStream(file, FileMode.Open)) using (var bytestream = new FileStream(file, FileMode.Open))
{ {
var magic = Read32(bytestream); var magic = Read32(bytestream);
@@ -118,6 +121,9 @@ namespace Tensorflow.Hub


private NDArray ExtractLabels(string file, bool one_hot = false, int num_classes = 10, int? limit = null) private NDArray ExtractLabels(string file, bool one_hot = false, int num_classes = 10, int? limit = null)
{ {
if (!Path.IsPathRooted(file))
file = Path.Combine(AppContext.BaseDirectory, file);
using (var bytestream = new FileStream(file, FileMode.Open)) using (var bytestream = new FileStream(file, FileMode.Open))
{ {
var magic = Read32(bytestream); var magic = Read32(bytestream);


+ 3
- 0
src/TensorFlowHub/Utils.cs View File

@@ -42,6 +42,9 @@ namespace Tensorflow.Hub


if (!Directory.Exists(saveTo)) if (!Directory.Exists(saveTo))
Directory.CreateDirectory(saveTo); Directory.CreateDirectory(saveTo);
if (!Path.IsPathRooted(zipFile))
zipFile = Path.Combine(AppContext.BaseDirectory, zipFile);


var destFilePath = Path.Combine(saveTo, Path.GetFileNameWithoutExtension(zipFile)); var destFilePath = Path.Combine(saveTo, Path.GetFileNameWithoutExtension(zipFile));




Loading…
Cancel
Save