부스트캠프 AI Tech 2기/2기 CV P-Stage-Semantic Segmentation

Competition Overview (EDA & Metric)

2021. 10. 18. 18:08
목차
  1. Recycle Trash 대회에 대한 Overview
  2. 1. COCO Dataset
  3. 2. EDA
  4. 3. 평가 Metric
  5. 4. Baseline Code
반응형

Recycle Trash 대회에 대한 Overview

1. COCO Dataset

1.1 COCO Format

- 모든 이미지를 뭉쳐서 3개의 batch로 나눴고, train valid test에 대한 구분은 json file로 구분되어져있다.

 

- 해당 영역이 어떤 class를 나타내는지 알려주기 때문에 channel이 필요가없다

1.2 DataLoader

1) data_dir: 데이터 셋 경로

2) mode

- train에 사용될지 test에 사용될지 구분 (mode="train", mode="test") 

- train일 경우 image_infos도 같이 return

3) ㅇtransform

- image size 조절 및 data format 변환 등의 전처리 작업

 

Train 모드 코드 설명

 

image_id = self.coco.getImgIds(imgIds=index)
image_infos = self.coco.loadImgs(image_id)[0]

>>>
image_id 
=> 0

image_infos 
=>
{'license':0,
 'url': None,
 'file_name': 'batch_01_vt/0003.jpg',
 'height': 512,
 'width': 512,
 'date_captured':None,
 'id':0}

 

cv2를 이용하여 image 불러오기

- 255로 나누어 정규화를 해주면 이미지가 제대로 보인다

images = cv2.imread(os.path.join(dataset_path, image_infos['file_name']))
images = cv2.cvtColor(images, cv2.COLOR_BGR2RGB).astype(np.float32)
images /= 255.0

 

시각화 할때에는

len(idx['segmentation'])[0]으로 sort를 하여 큰 순서대로 표시한다

background -> plastic bag -> generaltrash -> ... -> plastic bag

Transform

- Albumantations lib을 사용하면

ToTensorV2()를 사용하여  numpy (8 ,512, 512, 3)를 tensor (8, 3, 512, 512)로 변환해줘야한다

2. EDA

2.1 Image 분석

- 시각화 하여 확인해보자

2.2 class 분석

2.3 가설 검정

1) ‘Plastic’과 ‘Plastic bag’은 따로 라벨링 되었을까?

2) Plastic bag 내부의 재활용들도 라벨링이 되어있을까??

하지만 Plastic bag에 덮혀져있지 않은 부분들은 다른 클래스로 잘 분류되어있다

이미지에 돌멩이가 같이 labeling되어있는 것들은 이미지를 제거하거나 돌멩이를 background로 다시 labeling 하면 더 나은 성능을 기대해 볼 수 있다

 

 

3. 평가 Metric

3.1 mIoU

Mean IoU: 모두 구해진 IoU를(배경클래스를 포함) 평균낸다

4. Baseline Code

4.1 torchvision.models.segmentation.fcn_resnet50

- 마지막 부분 분류를 해당 클래스의 수에 맞게 변경해야한다.

4.2 loss / optimizer 설정

- Softmax + Cross-Entropy loss

 

반응형

'부스트캠프 AI Tech 2기 > 2기 CV P-Stage-Semantic Segmentation' 카테고리의 다른 글

Semantic Segmentation 대회에서 사용하는 방법들 1  (0) 2021.10.20
High Performance를 자랑하는 Unet 계열의 모델들  (0) 2021.10.20
FCN의 한계를 극복한 model 2 성능적인 측면에서의 극복  (0) 2021.10.20
Semantic Segmentation의 기초와 이해  (0) 2021.10.19
Semantic Segmentation - Intro  (0) 2021.10.18
  1. Recycle Trash 대회에 대한 Overview
  2. 1. COCO Dataset
  3. 2. EDA
  4. 3. 평가 Metric
  5. 4. Baseline Code
'부스트캠프 AI Tech 2기/2기 CV P-Stage-Semantic Segmentation' 카테고리의 다른 글
  • High Performance를 자랑하는 Unet 계열의 모델들
  • FCN의 한계를 극복한 model 2 성능적인 측면에서의 극복
  • Semantic Segmentation의 기초와 이해
  • Semantic Segmentation - Intro
모플로
모플로
현재 진행하는 개발 과정을 자주 기록하는 곳
반응형
모플로
모던플로우
모플로
전체
오늘
어제
  • 분류 전체보기 (143)
    • 개발 (41)
      • Python (5)
      • Docker (3)
      • Shell (3)
      • Airflow (0)
      • Elasticsearch (0)
      • AWS (4)
      • DataAnalysis (2)
      • Tensorflow (9)
      • NLP (4)
      • ML,DL 나중에 카테고리분류 (1)
      • Nestjs (1)
      • Git (4)
      • Algorithm (0)
      • 설계 (2)
      • Database (1)
    • 부스트캠프 AI Tech 2기 (97)
      • Precourse (27)
      • 2기 U-Stage (20)
      • 2기 P-Stage (2)
      • 2기 CV U-Stage (12)
      • 2기 CV P-Stage-Objectdetecti.. (10)
      • 2기 CV P-Stage-Semantic Segm.. (8)
      • 2기 CV 데이터제작 (7)
      • 2기 모델 최적화 (0)
      • 2기 Product Serving (10)

블로그 메뉴

  • 홈
  • 태그

공지사항

인기 글

태그

  • ubuntu20.04
  • transfer learning
  • detection
  • codepipeline
  • tensorflow
  • Docker
  • ElasticBeansTalk
  • Python
  • Semantic Segmentation
  • Node
  • 알고리즘
  • NestJS
  • Object
  • CV
  • fcn
  • aws
  • MLE
  • Transformer
  • nlp
  • efficientnet
  • MLOps
  • 경사하강법
  • 프로그래머스
  • Example
  • Object Detection
  • pytorch
  • CICD
  • gradient descent
  • yolo
  • ubuntu

최근 댓글

최근 글

hELLO · Designed By 정상우.v4.1.4
모플로
Competition Overview (EDA & Metric)
상단으로

티스토리툴바

단축키

내 블로그

내 블로그 - 관리자 홈 전환
Q
Q
새 글 쓰기
W
W

블로그 게시글

글 수정 (권한 있는 경우)
E
E
댓글 영역으로 이동
C
C

모든 영역

이 페이지의 URL 복사
S
S
맨 위로 이동
T
T
티스토리 홈 이동
H
H
단축키 안내
Shift + /
⇧ + /

* 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.