qurator-spk/sbb_pixelwise_segmentation
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git synced 2025-06-09 11:50:04 +02:00
Code Issues Releases Wiki Activity

continue training, losses and etc

Browse source
This commit is contained in:
vahid 2021-06-22 18:47:59 -04:00
parent 5fb7552dbe
commit 4bea9fd535
3 changed files with 69 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

2
utils.py
View file

@ -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 ) )