detectree2.tests package

Submodules

detectree2.tests.test_all module

detectree2.tests.test_data module

class detectree2.tests.test_data.TestCase(methodName='runTest')

Bases: TestCase

test_upper()

detectree2.tests.test_evaluation module

detectree2.tests.test_evaluation.test_site_f1_score()

Computes the F1 score on manufactured data from tests/input directory.

The data in the test and pred dirs contains manufactured data to test IoU It will give a zero value for one of the predictions and a value of 1 for the other. Small refactor of evaluation.py is needed. Improve evaluation functions. Function is currently incomplete.

detectree2.tests.test_evaluation.test_tree_feature()

Tests the Feature class with some dummy data from detectree2-data.

Function is incomplete. Only returns known crown_area. Crowns are boxes so this is easy to evaluate.

detectree2.tests.test_models module

class detectree2.tests.test_models.TestCase(methodName='runTest')

Bases: TestCase

test_detectron2()

detectree2.tests.test_predict module

class detectree2.tests.test_predict.TestCase(methodName='runTest')

Bases: TestCase

test_f1_cal()

test_prec_recall_func()

test_predict()

Test prediction function in test_predict.py.

Download pre-trained model and run inference step on known image used for training.

detectree2.tests.test_preprocessing module

Module contents