public void PrepareData()
{
// get model file
string url = "http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_coco_2018_01_28.tar.gz";
Web.Download(url, modelDir, "ssd_mobilenet_v1_coco.tar.gz");
Compress.ExtractTGZ(Path.Join(modelDir, "ssd_mobilenet_v1_coco.tar.gz"), "./");
// download sample picture
url = $"https://github.com/tensorflow/models/raw/master/research/object_detection/test_images/image2.jpg";
Web.Download(url, imageDir, "input.jpg");
// download the pbtxt file
url = $"https://raw.githubusercontent.com/tensorflow/models/master/research/object_detection/data/mscoco_label_map.pbtxt";
Web.Download(url, modelDir, "mscoco_label_map.pbtxt");
}
public override void PrepareData()
{
Directory.CreateDirectory(dir);
// get model file
string url = "https://storage.googleapis.com/download.tensorflow.org/models/inception_v3_2016_08_28_frozen.pb.tar.gz";
Web.Download(url, dir, $"{pbFile}.tar.gz");
Compress.ExtractTGZ(Path.Join(dir, $"{pbFile}.tar.gz"), dir);
// download sample picture
string pic = "grace_hopper.jpg";
url = $"https://raw.githubusercontent.com/tensorflow/tensorflow/master/tensorflow/examples/label_image/data/{pic}";
Web.Download(url, dir, pic);
}
public override void Train()
{
// get a set of images to teach the network about the new classes
string fileName = "flower_photos.tgz";
string dataDir = "image_classification_v1";
string url = $"http://download.tensorflow.org/example_images/{fileName}";
Web.Download(url, dataDir, fileName);
Compress.ExtractTGZ(Path.Join(dataDir, fileName), dataDir);
// using wizard to train model
var wizard = new ModelWizard();
var task = wizard.AddImageClassificationTask <TransferLearning>(new TaskOptions
{
DataDir = @"image_classification_v1\flower_photos",
});
task.Train(new TrainingOptions());
}
public void PrepareData()
{
// get a set of images to teach the network about the new classes
string fileName = "flower_photos.tgz";
string url = $"http://download.tensorflow.org/example_images/{fileName}";
Web.Download(url, data_dir, fileName);
Compress.ExtractTGZ(Path.Join(data_dir, fileName), data_dir);
// download graph meta data
url = "https://raw.githubusercontent.com/SciSharp/TensorFlow.NET/master/graph/InceptionV3.meta";
Web.Download(url, "graph", "InceptionV3.meta");
// download variables.data checkpoint file.
url = "https://github.com/SciSharp/TensorFlow.NET/raw/master/data/tfhub_modules.zip";
Web.Download(url, data_dir, "tfhub_modules.zip");
Compress.UnZip(Path.Join(data_dir, "tfhub_modules.zip"), "tfhub_modules");
// Prepare necessary directories that can be used during training
Directory.CreateDirectory(summaries_dir);
Directory.CreateDirectory(bottleneck_dir);
// Look at the folder structure, and create lists of all the images.
image_lists = create_image_lists();
class_count = len(image_lists);
if (class_count == 0)
{
print($"No valid folders of images found at {image_dir}");
}
if (class_count == 1)
{
print("Only one valid folder of images found at " +
image_dir +
" - multiple classes are needed for classification.");
}
}