mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-06-09 11:50:04 +02:00
continue training, losses and etc
This commit is contained in:
parent
5fb7552dbe
commit
4bea9fd535
3 changed files with 69 additions and 24 deletions
2
utils.py
2
utils.py
|
@ -374,9 +374,7 @@ def provide_patches(dir_img,dir_seg,dir_flow_train_imgs,
|
|||
|
||||
indexer=0
|
||||
for im, seg_i in tqdm(zip(imgs_cv_train,segs_cv_train)):
|
||||
#print(im, seg_i)
|
||||
img_name=im.split('.')[0]
|
||||
print(img_name,'img_name')
|
||||
if not patches:
|
||||
cv2.imwrite(dir_flow_train_imgs+'/img_'+str(indexer)+'.png', resize_image(cv2.imread(dir_img+'/'+im),input_height,input_width ) )
|
||||
cv2.imwrite(dir_flow_train_labels+'/img_'+str(indexer)+'.png' , resize_image(cv2.imread(dir_seg+'/'+img_name+'.png'),input_height,input_width ) )
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue