본문 바로가기
반응형

Study86

[PyTorch] torchvision을 활용한 image transform torchvision을 사용해서 이미지 변형하기 #이미지 가운데 부분을 자르는 것 torchvision.transforms.CenterCrop(size=(300, 300))(image) #범위 안에서 brightness 등 값이 랜덤하게 변함 torchvision.transforms.ColorJitter(brightness=1, contrast=0, saturation=0, hue=0)(image) #rgb를 gray scale로 변경 torchvision.transforms.Grayscale(num_output_channels=1)(image) #이미지 크기를 늘림 torchvision.transforms.Pad(padding=(20, 20), fill=0, padding_mode='constant').. 2021. 1. 31.
[tensorflow] 텐서플로우 콜백함수(Callbacks) 콜백함수란? 모델을 학습시키는 도중에 어떤 작업을 할 수 있게 하는데 그 작업을 callback이라고 한다. 텐서보드 열기 - 우선 저장위치를 지정한다. logdir=os.path.join('logs',datetime.now().strftime('%Y%m%d-%H%M%S')) tensorboard=tf.keras.callbacks.TensorBoard( log_dir=logdir, write_graph=True, write_images=True, #weight 등 시각화 여부 histogram_freq=1 ) LambdaCallback - TensorBoard에 customize한 이미지, 그래프를 넣을 수 있음 www.tensorflow.org/tensorboard/image_summaries#setu.. 2021. 1. 31.
[tensorflow] tf.data를 사용해 이미지 데이터 학습시키기 tf.data는 datagenerator를 사용하지 않는다. 이미지 가져오기 - 읽어온 이미지를 batch size만큼 묶어줘야 한다. - shuffle을 사용해 읽어온 이미지들의 순서를 섞어준다. dataset=tf.data.Dataset.from_tensor_slices(data_paths) dataset=dataset.map(read_image) dataset = dataset.batch(batch_size) dataset=dataset.shuffle(buffer_size=len(data_paths)) 이미지의 레이블 값 가져오기 def get_label(path): fname=tf.strings.split(path,'_')[-1] cls_name=tf.strings.regex_replace(fna.. 2021. 1. 25.
[tensorflow] dataframe 만들고 dataframe으로 학습시키기 DataFrame 생성하기 pandas 라이브러리가 csv, excel을 다룰 때 사용된다. DataFrame 생성 data_ex={'a':[1,2,3],'b':[10,20,30],'c':[100,200,300]} df_ex=pd.DataFrame(data_ex) df_ex a b c 0 1 10 100 1 2 20 200 2 3 30 300 다음과 같은 형태의 데이터 프레임이 생성된다. 생성한 데이터 프레임 저장하기 train_df.to_csv(train_csv_path, index=False) test_df.to_csv(test_csv_path, index=False) 데이터 프레임으로 학습 데이터 전처리 1. csv 데이터를 불러온다 test_df=pd.read_csv('test_dataset.cs.. 2021. 1. 25.
반응형

티스토리툴바