포디블록 구조 추출 AI 경진대회

[2 / 2 학습] Augmentation + Densenet201 + tta

2023.01.28 23:18 1,774 조회 language

다른 분들의 다양한 모델에 대한 소개도 참고할 수 있었고, 백그라운드를 이미지 합성하는 것도 효과가 좋다는 입증을 해주셨기에, 
앞의 데이터 합성 기법을 믿고 사용할 수 있었습니다.
또, timm에서 pretrained 데이터를 불러와 학습을 보다 빠르고 안정적으로 할 수 있도록 시도했습니다.
추론에서는 test time augmentation을 통해 결과에 대한 안정성을 늘려보려 했습니다..
학습에 있어서는 데이터 만들고, 학습하고, 만들고, 이어서 학습하고 반복했습니다.
학습에 사용할 pretrained model과 데이터 생성에 필요한 blender 파일을 첨부합니다.
https://drive.google.com/drive/folders/15UzucNTVaPEZwmwk7qylfiVI5u3euArs?usp=sharing
이 대회로 좋은 경험을 할 수 있었습니다. 감사합니다!

#학습 환경
CPU: 2x Intel(R) Xeon(R) Gold 5218R CPU @ 2.10GHz
intel(R) Core(TM) i9-12900K (24CPUs), ~3.2GHz
RAM: 64GB
GPU: NVIDIA RTX 3090
OS: Windows 10 HOME 64비트 (10.0, 빌드 19044)
PYTHON: 3.8
PYTORCH: torch 1.13.0+cu117
CUDA: 11.7

#Directory
├── open
│   ├── BLENDERMADE(제작된 이미지, 기존 train 이미지)
│   ├── BLENDERTRAIN
      │    ├── BLENDERMADE(합성 결과)
│   ├── background (배경)
│   ├── train  (배경합성이미지)
│   ├── test
│   ├── train_2.csv (블렌더 파일 포함 데이터)
│   ├── test.csv
│   ├── sample_submission.csv
│   ├── train.ipynb

PDF
코드
로그인이 필요합니다
0 / 1000
찌니찌니
2023.01.31 22:27

열정에 치얼스~

이세의인공지능
2023.01.31 22:44

뿌듯함과 교환한 왼쪽 손목입니다...ㅠ 3일 만에 드디어 잠을 잘 수 있겠네요!

찌니찌니
2023.02.02 12:26

이 대회 보면서 블록이라 블렌더나 opengl같은 걸로 3d 데이터 증강 가능하겠는데? 라는 생각만하고 설마... 누군가가 할까? 이생각 했는데 하신 분이 있어서 정말 놀랍고, 그것도 처음 블랜더를 사용하신분이라서 한번더 놀랐습니다!! 

이세의인공지능
2023.02.02 15:32

다들 노가다를 잘 안 좋아하시는 것 같아서 제가 시도해봤습니다 ㅋㅋ 점점 능숙해지니까 만드는게 빨라지더라구요

돼지와함께춤을
2023.02.11 00:18

이세님! 잘봤습니다!
공유해주신 블렌더 파일에 script가 없는거 같은데 확인 부탁드려도 될까요?

이세의인공지능
2023.02.11 01:08

와 관심가져주셔서 감사합니다!
more0026.blend 파일을 올렸습니다.
more0026.blend1에 텍스쳐 정보같은 부가 정보가 있는 것 같아요!
이 작업을 위해 공부한게 3시간 정도뿐이라 블렌더를 능숙히 다루진 못합니다.

돼지와함께춤을
2023.02.11 17:47

감사합니다. 이세님! 새해 복 많이 받으세요. ^_^

이세의인공지능
2023.02.11 17:35

감사합니다 ㅠㅠ 새해 복 많이 받으세요!!