cv2BGR

import cv2

img_path = '/media/wagnchogn/data_16t/segment data/PAIP/data/val_20x/01_01_0105_148_442_7168_7168_0_1024i0j.png'
img = cv2.imread(img_path)
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
cv2.imwrite('/media/wagnchogn/data_16t/segment data/PAIP/data/val_20x/01_01_0105_148_442_7168_7168_0_1024i0j_tttt.png', img)

import os
import glob
import shutil
from tqdm import tqdm
###1
# mask_route = r'/media/dell/data_4t/SAM_withmag/data/new-data/CoCaHis/crop_mask_v2'
# img_route =r'/media/dell/data_4t/SAM_withmag/data/new-data/CoCaHis/crop_img_v2'
#
# new_img_route = r'/media/dell/data_4t/SAM_withmag/data_mag_v2/cocahis/img'
# new_mask_route = r'/media/dell/data_4t/SAM_withmag/data_mag_v2/cocahis/mask'
#
# mask_route = r'/media/dell/data_4t/SAM_withmag/data_magpre/bcnb/crop/sm_10x_v2_name'
# img_route =r'/media/dell/data_4t/SAM_withmag/data_magpre/bcnb/crop/imgs_10x'
#
# new_img_route = r'/media/dell/data_4t/SAM_withmag/data_mag_v2/data_0401/bcnc_10x/img'
# new_mask_route = r'/media/dell/data_4t/SAM_withmag/data_mag_v2/data_0401/bcnb_10x/mask'


mask_route = r'/media/wagnchogn/data_16t/segment data/WBC/resize/dataset_1/masks'
img_route =r'/media/wagnchogn/data_16t/segment data/WBC/resize/dataset_1/imgs'

new_img_route = r'/media/wagnchogn/data_16t/data_json/WBC/img'
new_mask_route = r'/media/wagnchogn/data_16t/data_json/WBC/mask'

if not os.path.exists(new_mask_route):
    os.makedirs(new_mask_route)
if not os.path.exists(new_img_route):
    os.makedirs(new_img_route)

# img_files = os.listdir(img_route)
img_files = [f for f in os.listdir(img_route) if f.endswith('.png')]

for img_file in tqdm(img_files):
    img_name = img_file.split('.png')[0]
    mask_files = glob.glob(os.path.join(mask_route, img_name + '_' +'*.png'))
    if len(mask_files) == 0:
        continue
    shutil.copy(os.path.join(img_route, img_file), new_img_route)
    new_mask_path = os.path.join(new_mask_route, img_name)
    if not os.path.exists(new_mask_path):
        os.makedirs(new_mask_path)
    for mask_file in mask_files:
        shutil.copy(os.path.join(mask_route, mask_file), new_mask_path)
    # print(len(mask_files))

print('move done')

import os
import cv2
from tqdm import tqdm



# data_route = '/root/autodl-tmp/dataset/256_test_code/tnbc/masks'
data_route = r'/media/dell/data_4t/SAM_withmag/data_mag_v2/data_0401/glas_20x/mask'
img_files = os.listdir(data_route)
for img_file in tqdm(img_files):
    img_path = os.path.join(data_route, img_file)
    mask_files = os.listdir(img_path)
    for mask_file in mask_files:
        mask_path = os.path.join(img_path, mask_file)
        mask = cv2.imread(mask_path, cv2.IMREAD_GRAYSCALE)
        mask[(mask != 0) & (mask != 255)] = 255
        cv2.imwrite(mask_path, mask)
print("check done")

import os
import json
import os

def load_json(path):
    with open(path, 'r') as f:
        data = json.load(f)
    return data

saved_file = '/media/dell/data_4t/SAM_withmag/data_mag_v2/data_0401/glas_20x'

train_label_path = os.path.join(saved_file, "image2label_train.json")
train_mag_path = os.path.join(saved_file, 'image2mag_train.json')
test_label_path = os.path.join(saved_file, 'label2image_test.json')
test_mag_path = os.path.join(saved_file, 'label2mag_test.json')

train_label_dict = load_json(train_label_path)
train_mag_dict = load_json(train_mag_path)
test_label_dict = load_json(test_label_path)
test_mag_dict = load_json(test_mag_path)

train_labels = list(train_label_dict.keys())
train_mags = list(train_mag_dict.keys())

test_labels = list(test_label_dict.keys())
test_mags = list(test_mag_dict.keys())

print(train_labels==train_mags)
print(test_labels==test_mags)