티스토리 뷰

파이썬으로 빈 txt파일 만들기

딥러닝에서 dummy파일을 만들기 위해서 구글 크롤링을 해서 사진을 뽑아 왔다.

근데 크롬에서 크롤링이 막혀

저번에 다운받은 이미지로 진행했다.

그리고 resize를 이용해서 크기를 바꿨다. 이후 번호가 1부터 시작하는게 아니라 무작위 숫자로 되있어서 다 정렬을 새롭게 했다.

이후 필요한 빈 lable txt파일을 생성했다. 사진 개수만큼


 

밑에 블로그에서 참고한 resize하는 법

ponyozzang.tistory.com/600

#!usr/bin/env python
# -*- coding: utf-8 -*-


import os
import glob
from PIL import Image

files = glob.glob('/home/sy/dummy/2021-03-17/wall texture2/*')

for f in files:
    try:
        img = Image.open(f)
        img_resize = img.resize((1920,1080))
        title, ext = os.path.splitext(f)
        img_resize.save(title + ext)
    except OSError as e:
        pass



print("resize")

전체 파일이름 한번에 바꾸기

이미지가 이미 숫자로 되어있어서 (ex: 1.jpg) 구글링한걸 썼을 때 이상하게 됐었는데

이름을 문자랑 같이 있는걸로 만들어주고 새롭게 1부터 시작하게 만들었더니 오류가 없어졌다.

왜그런지 모른다. 일단 기록용이라서,,ㅎ

 

구글링한 블로그

hogni.tistory.com/35

#!usr/bin/env python
# -*- coding: utf-8 -*-  

import os
import glob

files = glob.glob('/home/sy/dummy/2021-03-17/wall texture2/*')

# 타이틀+숫자.jpg로 변경
for i, f in enumerate(files):
    ftitle, fext = os.path.splitext(f)
    os.rename(f, ftitle + '_' + '{0:03d}'.format(i) + fext)


# 숫자.jpg로 변경
file_path = '/home/sy/dummy/2021-03-17/wall texture2' #이미지가 저장된 경로
file_names = os.listdir(file_path)
file_names

i = 1
for name in file_names:
    src = os.path.join(file_path, name)
    dst = str(i) + '.jpg'
    dst = os.path.join(file_path, dst)
    os.rename(src, dst)
    i += 1

이미 파일이름이 숫자랑 문자랑 같이 된 파일이면 윗부분 타이틀+숫자.jpg로 만들어주는 부분은 필요없다.

 

label txt 만들기

여러개의 빈 txt 파일을 만들기 도전

for문을 이용해서 반복적으로 해봤다.

#!usr/bin/env python
# -*- coding: utf-8 -*-

import os
import glob

file_path = '/home/sy/dummy/2021-03-17/wall texture2'
file_names = os.listdir(file_path)
file_names

i = 1
for name in file_names:
    f = open(str(i)+'.txt', 'w')
    f.close()
    i += 1

이 코드는 내가 스스로 만든것! 열심히 python공부중 ㅜㅜ

open함수를 이용해서 txt파일을 생성했다.

 

 

 

사담)

그럼 더 파이썬을 공부하고 오겠다...

도전과제!

이제 만든 데이터셋으로 객체 인식을 도전할거다 > <

 

 

댓글