updated tests
This commit is contained in:
@@ -0,0 +1,13 @@
|
||||
from src.database import ModelOutputs
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# instantiate data object
|
||||
annotation = {
|
||||
|
||||
}
|
||||
vis_com_list = [ModelOutputs.from_annotation(path) for path in img_path_list]
|
||||
# generate random predictions
|
||||
[vis_com.generate_random_prediction() for vis_com in vis_com_list]
|
||||
for vis_com in vis_com_list:
|
||||
print(vis_com)
|
||||
Reference in New Issue
Block a user