pytorch 사용하시는분들 transform 관련 소소한 팁

대학 대항전 : 퍼즐 이미지 AI 경진대회

귀여미

2024.01.03 17:41 2,657 Views

transform 대신에 transform.v2 사용해 보세요

베타버전지만 속도 향상이 있다고 하네요

colorjitter나 augmix등등 무거운 전처리는 약 10%의 속도 향상이 있었습니다..

(colorjitter 사용, 배치사이즈 16 환경, num_workers=4, train it/s 36 -> 40로 향상)

basline code의 transform예시입니다

train_transform = transforms.v2.Compose([

v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]),

v2.Resize((CFG['IMG_SIZE'], CFG['IMG_SIZE'])),

v2.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])

])

test_transform = transforms.v2.Compose([

v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]),

v2.Resize((CFG['IMG_SIZE'], CFG['IMG_SIZE'])),

v2.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])

])

https://pytorch.org/vision/0.15/transforms.html

https://pytorch.org/vision/0.15/auto_examples/plot_transforms_v2.html#sphx-glr-auto-examples-plot-transforms-v2-py

2 Comments

comment

0 / 1000

JAEM

2024.01.03 23:21

image = Image.open(img_path).convert("RGB")를
image = cv2.imread(img_path)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
로 바꿔주는것도 속도 향상에 도움이 됩니다

귀여미님이 말씀해주신 transform.v2나 albumentation이랑 같이 사용하면 좋을 것 같네요

귀여미

2024.01.04 15:44

좋은 정보 감사합니다!!

📣 The comment input field has been moved to the top of the comment list!

List

2차 평가 관련 문의입니다

Competition - 2023 AICOSS 해커톤 경진대회

Current