22 lines
608 B
Python
22 lines
608 B
Python
from pathlib import Path
|
|
|
|
from src.database import VisualCommunication
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# get list of image paths
|
|
test_dir = Path(__file__).parent
|
|
img_dir = test_dir / "imgs"
|
|
img_path_list = [path for path in img_dir.glob("*.jpeg") if path.is_file()]
|
|
print(img_path_list)
|
|
# instantiate data object
|
|
vis_com_list = [
|
|
VisualCommunication.from_file(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)
|