def test_fastrcnn_grocery_training(device_id):
if cntk_device(device_id).type() != DeviceKind_GPU:
pytest.skip('test only runs on GPU') # it runs very slow in CPU
try_set_default_device(cntk_device(device_id))
from A1_GenerateInputROIs import generate_input_rois
assert generate_input_rois(testing=True)
# since we do not use a reader for evaluation we need unzipped data
externalData = 'CNTK_EXTERNAL_TESTDATA_SOURCE_DIRECTORY' in os.environ
if externalData:
extPath = os.environ['CNTK_EXTERNAL_TESTDATA_SOURCE_DIRECTORY']
model_file = os.path.join(extPath, "PreTrainedModels", "AlexNet", "v0",
"AlexNet.model")
else:
model_file = os.path.join(
abs_path, *"../../../../PretrainedModels/AlexNet.model".split("/"))
from A2_RunWithPyModel import train_fast_rcnn, evaluate_fast_rcnn
trained_model = train_fast_rcnn(model_path=model_file)
assert evaluate_fast_rcnn(trained_model)
from A3_ParseAndEvaluateOutput import evaluate_output
assert evaluate_output()
from B3_VisualizeOutputROIs import visualize_output_rois
assert visualize_output_rois(testing=True)
def test_fastrcnn_grocery_training(device_id):
if cntk_device(device_id).type() != DeviceKind_GPU:
pytest.skip('test only runs on GPU') # it runs very slow in CPU
try_set_default_device(cntk_device(device_id))
from A1_GenerateInputROIs import generate_input_rois
assert generate_input_rois(testing=True)
# since we do not use a reader for evaluation we need unzipped data
externalData = 'CNTK_EXTERNAL_TESTDATA_SOURCE_DIRECTORY' in os.environ
if externalData:
extPath = os.environ['CNTK_EXTERNAL_TESTDATA_SOURCE_DIRECTORY']
model_file = os.path.join(extPath, "PreTrainedModels", "AlexNet", "v0", "AlexNet.model")
else:
model_file = os.path.join(abs_path, *"../../../../Examples/Image/PretrainedModels/AlexNet.model".split("/"))
from A2_RunWithPyModel import train_fast_rcnn, evaluate_fast_rcnn
trained_model = train_fast_rcnn(model_path=model_file)
assert evaluate_fast_rcnn(trained_model)
from A3_ParseAndEvaluateOutput import evaluate_output
assert evaluate_output()
from B3_VisualizeOutputROIs import visualize_output_rois
assert visualize_output_rois(testing=True)
def test_fastrcnn_grocery_visualization():
from A1_GenerateInputROIs import generate_input_rois
assert generate_input_rois(testing=True)
from B1_VisualizeInputROIs import generate_rois_visualization
assert generate_rois_visualization(testing=True)
from B2_EvaluateInputROIs import evaluate_rois
assert evaluate_rois()
def test_fastrcnn_grocery_visualization():
from A1_GenerateInputROIs import generate_input_rois
assert generate_input_rois(testing=True)
from B1_VisualizeInputROIs import generate_rois_visualization
assert generate_rois_visualization(testing=True)
from B2_EvaluateInputROIs import evaluate_rois
assert evaluate_rois()
def test_fastrcnn_with_config_file(device_id):
if cntk_device(device_id).type() != DeviceKind_GPU:
pytest.skip('test only runs on GPU') # it runs very slow in CPU
try_set_default_device(cntk_device(device_id))
from A1_GenerateInputROIs import generate_input_rois
assert generate_input_rois(testing=True)
prepare_alexnet_v0_model()
from A2_RunWithBSModel import run_fastrcnn_with_config_file
assert run_fastrcnn_with_config_file(os.environ["TEST_CNTK_BINARY"])
def test_fastrcnn_with_config_file(device_id):
if cntk_device(device_id).type() != DeviceKind_GPU:
pytest.skip('test only runs on GPU') # it runs very slow in CPU
try_set_default_device(cntk_device(device_id))
from A1_GenerateInputROIs import generate_input_rois
assert generate_input_rois(testing=True)
prepare_alexnet_v0_model()
from A2_RunWithBSModel import run_fastrcnn_with_config_file
assert run_fastrcnn_with_config_file(os.environ["TEST_CNTK_BINARY"])