jhynsoo/worlabel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'ai/refactor/resource-cleanup' into 'ai/develop'

Browse Source 
Refactor: process_image_and_label url입출력 로직 수정

See merge request s11-s-project/S11P21S002!280
This commit is contained in:
김용수 2024-10-04 13:03:07 +09:00
commit 4a626ab050
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ai/app/utils/file_utils.py
View File

@ -56,7 +56,8 @@ def process_image_and_label(data:TrainDataInfo, dataset_root_path:str, child_pat
label_path = os.path.join(dataset_root_path, child_path, f"{img_title}.txt")
# 레이블 객체 불러오기
label = json.loads(urllib.request.urlopen(data.data_url).read())
with urllib.request.urlopen(data.data_url) as response:
label = json.loads(response.read())
# 레이블 -> 학습용 레이블 데이터 파싱 후 생성
if label['task_type'] == "det":
@ -139,7 +140,8 @@ def process_image_and_label_in_cls(data:TrainDataInfo, dataset_root_path:str, ch
img_name = data.image_url.split('/')[-1]
# 레이블 객체 불러오기
label = json.loads(urllib.request.urlopen(data.data_url).read())
with urllib.request.urlopen(data.data_url) as response:
label = json.loads(response.read())
if not label["shapes"]:
# assert label["shapes"], No Label. Failed Download" # AssertionError 발생