added codes

main
pinb 7 months ago
parent c64abc4206
commit 3e98c2ad1c

@ -0,0 +1,54 @@
import os
from glob import glob
img_dir = 'C:/Users/pinb/Desktop/imgs'
mask_dir = 'C:/Users/pinb/Desktop/masks'
names = ['gold_ii',
'gold_ie',
'gold_io',
'gold_ee',
'gold_ei',
'gold_eo',
'gold_oi',
'gold_oe',
'gold_oo',
'silver_ii',
'silver_ie',
'silver_io',
'silver_ee',
'silver_ei',
'silver_eo',
'silver_oi',
'silver_oe',
'silver_oo']
list_imgs = glob(os.path.join(img_dir, '**', '*.png'), recursive=True)
list_masks = glob(os.path.join(mask_dir, '**', '*.png'), recursive=True)
remove_file = []
# 마스크 파일 이름 목록 생성 (확장자 제거 및 '_mask' 추가)
list_imgs_basenames = set([x.replace('_mask', '').replace(mask_dir, img_dir) for x in list_masks])
# list_imgs에서 list_masks에 없는 파일 찾기
imgs_to_delete = [mask for mask in list_imgs_basenames if mask not in list_imgs]
print(len(list_imgs))
print(len(list_masks))
print(len(imgs_to_delete))
print(imgs_to_delete)
masks_to_delete = set([x.replace('.png', '_mask.png').replace(img_dir, mask_dir) for x in imgs_to_delete])
for mask in masks_to_delete:
os.remove(mask)
print(f'Deleted: {mask}')
list_imgs = glob(os.path.join(img_dir, '**', '*.png'), recursive=True)
list_masks = glob(os.path.join(mask_dir, '**', '*.png'), recursive=True)
print(len(list_imgs))
print(len(list_masks))

@ -0,0 +1,31 @@
import torch
print("PyTorch 버전:", torch.__version__)
# CUDA 사용 가능 여부 확인
if torch.cuda.is_available():
print("CUDA is available! Testing CUDA...")
# CUDA 디바이스 설정
device = torch.device("cuda")
# CUDA 디바이스에 텐서를 생성하고 연산 수행
x = torch.rand(5, 5, device=device)
y = torch.rand(5, 5, device=device)
z = x + y
# 결과 출력
print("Successfully performed a CUDA operation:")
print(z)
else:
print("CUDA is not available.")
# PyTorch 버전 확인
print("PyTorch version:", torch.__version__)
# CUDA 사용 가능 여부 및 버전 확인
cuda_available = torch.cuda.is_available()
print("CUDA available:", cuda_available)
if cuda_available:
print("CUDA version:", torch.version.cuda)

Binary file not shown.

After

Width:  |  Height:  |  Size: 28 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 28 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 33 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 35 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 17 KiB

@ -0,0 +1,8 @@
Data types: int32 int32
Shapes: (262144,) (262144,)
Label sample: [0 0 0 ... 0 0 0]
Output sample: [0 0 0 ... 0 0 0]
Accuracy: 0.9977531433105469
Precision: 0.9608572440665958
Recall: 0.9866327180140039
F1 Score: 0.9735744089012517

Binary file not shown.

After

Width:  |  Height:  |  Size: 28 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 39 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 40 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

@ -0,0 +1,8 @@
Data types: int32 int32
Shapes: (262144,) (262144,)
Label sample: [0 0 0 ... 0 0 0]
Output sample: [0 0 0 ... 0 0 0]
Accuracy: 0.9964599609375
Precision: 0.9360505166475316
Recall: 0.9564758329422807
F1 Score: 0.9461529534640826

Binary file not shown.

After

Width:  |  Height:  |  Size: 28 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 24 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 36 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

@ -0,0 +1,8 @@
Data types: int32 int32
Shapes: (262144,) (262144,)
Label sample: [0 0 0 ... 0 0 0]
Output sample: [0 0 0 ... 0 0 0]
Accuracy: 0.9957275390625
Precision: 0.9423006583760059
Recall: 0.9546970539188439
F1 Score: 0.9484583525080534

Binary file not shown.

After

Width:  |  Height:  |  Size: 28 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 28 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 32 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 16 KiB

@ -0,0 +1,8 @@
Data types: int32 int32
Shapes: (262144,) (262144,)
Label sample: [0 0 0 ... 0 0 0]
Output sample: [0 0 0 ... 0 0 0]
Accuracy: 0.9977951049804688
Precision: 0.9486012970670796
Recall: 0.9952269726820351
F1 Score: 0.9713549410248785

Binary file not shown.

After

Width:  |  Height:  |  Size: 28 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 36 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 31 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 16 KiB

@ -0,0 +1,8 @@
Data types: int32 int32
Shapes: (262144,) (262144,)
Label sample: [0 0 0 ... 0 0 0]
Output sample: [0 0 0 ... 0 0 0]
Accuracy: 0.998199462890625
Precision: 0.9588396707173658
Recall: 0.9947132980886539
F1 Score: 0.9764471057884232

Binary file not shown.

After

Width:  |  Height:  |  Size: 30 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 206 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 28 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 39 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 36 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 17 KiB

@ -0,0 +1,8 @@
Data types: int32 int32
Shapes: (262144,) (262144,)
Label sample: [0 0 0 ... 0 0 0]
Output sample: [0 0 0 ... 0 0 0]
Accuracy: 0.9981002807617188
Precision: 0.9655700766110861
Recall: 0.9892880229014683
F1 Score: 0.9772851669403393

Binary file not shown.

After

Width:  |  Height:  |  Size: 28 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 25 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 33 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

@ -0,0 +1,8 @@
Data types: int32 int32
Shapes: (262144,) (262144,)
Label sample: [0 0 0 ... 0 0 0]
Output sample: [0 0 0 ... 0 0 0]
Accuracy: 0.9965438842773438
Precision: 0.947846012832264
Recall: 0.9684397827308485
F1 Score: 0.9580322401334075

Binary file not shown.

After

Width:  |  Height:  |  Size: 36 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 62 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 34 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 17 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 21 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 21 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 21 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 20 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 20 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 20 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 22 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 22 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 22 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 22 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 22 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 21 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 24 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 27 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 30 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 40 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 17 KiB

@ -0,0 +1,8 @@
Data types: int32 int32
Shapes: (262144,) (262144,)
Label sample: [0 0 0 ... 0 0 0]
Output sample: [0 0 0 ... 0 0 0]
Accuracy: 0.9976158142089844
Precision: 0.9365956036672927
Recall: 0.994021101992966
F1 Score: 0.9644543024512312

Binary file not shown.

After

Width:  |  Height:  |  Size: 28 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 30 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 206 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 34 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 36 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 62 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 17 KiB

@ -0,0 +1,4 @@
Accuracy: 0.9975090026855469
Precision: 0.9646868359809775
Recall: 0.9690994104492783
F1 Score: 0.9668880888393083

@ -0,0 +1,47 @@
import os
import shutil
from PIL import Image
import cv2
image_path = 'C:/Users/pinb/Desktop/image/'
mask_path = 'C:/Users/pinb/Desktop/mask/'
# names = ['gold_ii',
# 'gold_ie',
# 'gold_io',
# 'gold_ee',
# 'gold_ei',
# 'gold_eo',
# 'gold_oi',
# 'gold_oe',
# 'gold_oo',
names = ['silver_ii',
'silver_ie',
'silver_io',
'silver_ee',
'silver_ei',
'silver_eo',
'silver_oi',
'silver_oe',
'silver_oo']
for name in names:
image_forder = image_path + name
mask_forder = mask_path + name
for dir in os.listdir(image_forder):
image_bar_folder = os.path.join(image_forder, dir)
for file in os.listdir(image_bar_folder):
new_filename = os.path.join(image_bar_folder, file)
png_filename = os.path.splitext(os.path.basename(file))[0] + '.png'
new_png_file = os.path.join(image_bar_folder, png_filename)
img = cv2.imread(new_filename, cv2.IMREAD_GRAYSCALE)
cv2.imwrite(new_png_file, img)
os.remove(new_filename)
print(f'Converted "{new_filename}" to "{new_png_file}"')
# with Image.open(new_filename) as img:
# # PNG로 저장
# img.save(new_png_file, 'PNG')
# img.close()
# os.remove(new_filename)
# print(f'Converted "{new_filename}" to "{new_png_file}"')

@ -0,0 +1,86 @@
import os
import shutil
from PIL import Image
import cv2
# # 변환하고자 하는 폴더의 경로
# folder_path = 'C:/Users/pinb/Desktop/image/silver_oe/'
# # 변경하고자 하는 문자열과 변경될 문자열
# old_string = 'ie'
# new_string = 'oe'
# # 폴더 내의 모든 파일에 대해 반복
# for dir in os.listdir(folder_path):
# bar_folder = os.path.join(os.path.dirname(folder_path), dir)
# for filename in os.listdir(bar_folder):
# if old_string in filename:
# # 파일의 새 이름 생성
# new_filename = filename.replace(old_string, new_string)
# # 파일 이름 변경
# old_file_path = os.path.join(bar_folder, filename)
# new_file_path = os.path.join(bar_folder, new_filename)
# # 파일 이름 변경 실행
# os.rename(old_file_path, new_file_path)
# print(f'Renamed "{filename}" to "{new_filename}"')
image_path = 'C:/Users/pinb/Desktop/image/'
mask_path = 'C:/Users/pinb/Desktop/mask/'
#names = ['gold_ii',
# 'gold_ie',
# 'gold_io',
# 'gold_ee',
# 'gold_ei',
# 'gold_eo',
# 'gold_oi',
names = ['gold_oe',
'gold_oo']
# 'silver_ii',
# 'silver_ie',
# 'silver_io',
# 'silver_ee',
# 'silver_ei',
# 'silver_eo',
# 'silver_oi',
# 'silver_oe',
# 'silver_oo']
for name in names:
image_forder = image_path + name
# mask_forder = mask_path + name
# mask_origin = mask_path + name + '_mask.png'
for dir in os.listdir(image_forder):
image_bar_folder = os.path.join(image_forder, dir)
# mask_bar_folder = os.path.join(mask_forder, dir)
# if not os.path.exists(mask_bar_folder):
# os.makedirs(mask_bar_folder)
for file in os.listdir(image_bar_folder):
# filename = os.path.splitext(os.path.basename(file))[0] + '_mask.png'
# new_filename = os.path.join(mask_bar_folder, filename)
# shutil.copy2(mask_origin, new_filename)
# print(f'Copied "{mask_origin}" to "{new_filename}"')
new_filename = os.path.join(image_bar_folder, file)
png_filename = os.path.splitext(os.path.basename(file))[0] + '.png'
new_png_file = os.path.join(image_bar_folder, png_filename)
# with Image.open(new_filename) as img:
# # PNG로 저장
# img.save(new_png_file, 'PNG')
# img.close()
# os.remove(new_filename)
# print(f'Converted "{new_filename}" to "{new_png_file}"')
try:
split = os.path.splitext(os.path.basename(file))[1]
if split == '.bmp':
img = cv2.imread(new_filename, cv2.IMREAD_GRAYSCALE)
cv2.imwrite(new_png_file, img)
os.remove(new_filename)
print(f'Converted "{new_filename}" to "{new_png_file}"')
except:
os.remove(new_filename)
pass

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long
Loading…
Cancel
Save