병변 검출 AI 경진대회

알고리즘 | CV | 분류 | 검출 | mAP

상금 : 총 1,000만원
803명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

Barcelona팀, Private 3위, Private 점수: 0.82661, 앙상블

Barcelona

공동작성자

2021.12.09 20:53 7,443 조회 language

다들 대회 참여하느라 고생하셨습니다.
YoloV5와 MMDetection 두 개의 Framework를 사용하여 코드가 너무 많아 shell 파일 위주로 작성하였습니다.
자세한 코드는 Github를 통해 확인해주시면 감사하겠습니다. 그리고 문의사항이나 코드 오류가 있으면 편하게 말씀해주세요.
Github url: https://github.com/wooseok-shin/Lesion-Detection-3rd-place-solution

코드

댓글 10개

로그인이 필요합니다

comment

0 / 1000

쥐

2021.12.11 22:14

자극 받고 갑니다 바셀님

Barcelona

2021.12.13 15:32

화이팅입니다~ ㅎㅎ

하르딘

2021.12.12 17:51

안녕하세요 좋은 코드 감사합니다.
혹시 tta를 적용할지 안 할지는 yolov5, mmdetection 각각에서 어떤 코드가 제어하는지 여쭈어 봐도 될까요?

Barcelona

2021.12.13 15:31

YoloV5는 python detect.py 에서 --augment 라는 인자를 넣어주면 True가 되면서 TTA가 실행됩니다.
mmdetection은 제 github에서 보시면 mmdet_configs라는 폴더에 각 모델의 Config py 파일이 있는데 centernet을 기준으로하면 exp8_~~.py 파일에서 test_pipeline에서 flip인자를 True로 주시면 됩니다. 그리고 type에 MultiScaleFlipAug라고 되어있는데, img_scale=[(1333, 400), (1333, 800)] 라는 식으로 인자값을 주면 해당 값들을 기준으로 Multi-scale 이미지 사이즈 TTA를 수행합니다.
test_pipeline = [
    dict(type='LoadImageFromFile', to_float32=True),
    dict(
        type='MultiScaleFlipAug',
        scale_factor=1.0,
        flip=False,  # <-- True=TTA
        transforms=[
        .....

하르딘

2021.12.14 16:15

아 그렇군요 감사합니다.

Hwanython

2022.01.06 13:29

안녕하세요!!
좋은 정보 공유해주셔서 많이 배웁니다.

제가 굉장히 인공지능초보자로 차근차근 해보고 있는데요, 이 과정 진행하실때 혹시 하드웨어 환경, gpu 가 무엇이였을까요?

mmdet tool의 bash 사용 하실때는 gpu 옵션을 1로 두신거 같은데 제가 이해한게 맞을지요?

Barcelona

2022.01.07 16:41

네, 제대로 이해하신게 맞습니다!
CPU는 AMD Ryzen 3950이고, GPU는 RTX 3090를 사용했습니다.
제가 총 학습에 사용한 컴퓨터는 2대였고, 각 컴퓨터에 3090 gpu가 하나씩있어서 gpu 개수 옵션을 1로 하여 Single gpu 학습을 하였습니다!

Hwanython

2022.01.13 09:41

친절한 답변 감사드립니다!!! 많이 배웁니다.