viktor/wrongmove
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

detect floorplan using asyncio

Browse source
This commit is contained in:
Viktor Barzin 2025-05-17 22:58:35 +00:00
parent 68cc70bd11
commit b1e0ed170b
No known key found for this signature in database
GPG key ID: 4056458DBDBF8863
3 changed files with 22 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

7
crawler/data_access.py
View file

@ -1,3 +1,4 @@
import asyncio
from dataclasses import dataclass
import json
import pathlib
@ -135,13 +136,15 @@ class Listing:
) # filter out Nones
return max_sqm
def calculate_sqm_ocr(self, recalculate=True):
async def calculate_sqm_ocr(self, recalculate=True):
if not recalculate and self.path_floorplan_ocr_json().exists():
return
objs = []
for floorplan_path in self.list_floorplans():
estimated_sqm, model_output = floorplan.calculate_ocr(floorplan_path)
estimated_sqm, model_output = await asyncio.to_thread(
floorplan.calculate_ocr, floorplan_path
)
objs.append(
{
"floorplan_path": str(floorplan_path),