농업 환경 변화에 따른 작물 병해 진단 AI 경진대회

object detection task 로 바꿔 pretrain 모델을 만들고 있습니다.

2022.01.27 12:41 3,111 조회

BBox 를 활용하여 object detection task 로 만들어주고 모델에서 backbone 만 빼서 저희대회 데이터에 다시 finetuning 할 생각을 하고 있습니다.

이렇게 하면 Bbox 를 통해 모델이 주목해야 할 곳을 가이드해주는 효과가 있을거라 기대합니다.


아래는 기본제공 모델(Resnet50 backbone + fasterrcnn+ fpn) 의 결과인데, SOTA 모델 활용 + downstream task 에서의 결과가 잘 나오면 정리해서 한 번 더 올리도록 하겠습니다.

이번에 이렇게 글을 올린 것은 혹시 저와 비슷한 접근을 하시는 분들이 계시다면 의견을 여쭙고 싶어서입니다.


mmdetection 을 사용해 작업을 진행하는데, 최신 모델들은 전부 maskrcnn config 를 제공해서 저희 task 에 사용을 못하는 점이 아쉽습니다.

제가 사용하고 싶은 모델인 ConvNext 모델도 maskrcnn config 만 제공합니다.

제가 object detection 에 대해 잘 몰라서 그런데, 혹시 maskrcnn 모델도 저희 task 에 활용할 수 있을까요..?



===================================================================================================================

하.. 아직은 별로 효과가 없네요..

로그인이 필요합니다
0 / 1000
yh9468
2022.01.27 15:01

제가 알기론 maskrcnn은 segmentation정보를 사용하기 때문에 사용이 불가능한것으로 알고 있습니다

이외에도 다른 디텍션 모델을 사용하시는게 어떠신가요?

클로드섀넌
2022.01.27 19:26

https://github.com/open-mmlab/mmdetection

여기서 지원하는 모델들 중 하나 사용하려 하는데, 추천하시는 모델이 있으실까요?!

IAB
2022.01.27 15:47

Detection을 사용하여 이미지 전처리를 진행하는 경우, 정보 손실을 최소화하면서 Object 위주의 데이터 확보가 가능할 것 같긴하네요. 한 가지 우려되는 점은 Detection을 활용한 전처리에서의 추론 시간이 상당할 것으로 예상됩니다. 이 부분에 대해서는 성능 향상과 추론시간의 trade-off에서 적절성에 대한 판단이 필요할 것 같습니다.

클로드섀넌
2022.01.27 19:24

backbone 만 뽑아다 쓸거라 추론시간은 기존 resnet50 과 같을겁니다!

전처리 모듈 처럼 활용하는 방안도 있겠군요.. 하지만, 이건 말씀하신대로 너무 오래걸릴 것 같아 하지 않겠습니다.. ㅋㅋㅋ

클로드섀넌
2022.01.28 09:45

헉, 선생님 지금 1등이셨군요..
혹시 괜찮으시다면 성능향상을 위한 작은 팁이라도 얻을 수 있을까요..?
몸은 달아있는데 되는 게 없으니 답답합니다 ㅠㅠ 


IAB
2022.01.29 11:31

일단 간략하게 말씀드릴 수 있는건 데이터에 알맞는 튜닝 전략인 것 같습니다 ㅠㅠ 
추가적으로 공유할 수 있는 사항 정리해서 추후 토크란에 올려볼 수 있도록 노력해보겠습니다.

클로드섀넌
2022.01.29 23:15

저희가 받은 데이터의 가장 큰 특징이라면 같은 질병에서도 risk 정도를 나눠야 한다는 것 같은데, 이 risk 를 잘 구별하기 위한 튜닝전략을 준비해야 된다로 이해해도 될까요..?

IAB
2022.01.30 10:23

네, 맞는것 같습니다.

funcoolsexy
2022.01.27 16:27

윗분 말씀처럼 정보손실 최소화 측면에서 좋은 접근인 것 같습니다. SOTA모델의 성능과 추론시간이 궁금하네요. 추후에 공유해주시면 감사하겠습니다. 
현재 사용하신 Resnet50 backbone + fasterrcnn+ fpn을 통한 추론 시간은 대략 어느정도 나오는지 알 수 있을까요?