Code Monkey home page Code Monkey logo

liver-lesions-detection-2023's Introduction

Liver Lesions Detection based on Ultrasound Image Hackathon

This repository presents the 1st place solution for the Liver Lesions Detection based on Ultrasound Image Hackathon held from August 12th to 14th, 2023, at Mahidol University.

The solution leverages the DINO architecture with a Swin Transformer backbone to detect three classes of liver lesions (cystic, fibrosis, solid) as well as normal liver images. Our solution achieved a mAP70 score of 0.5037 on the public leaderboard (using 25% of the test data) and 0.5263 on the private leaderboard (utilizing the entire test data).

Qualitative Results

Qualitative Results

Our Observations

  • The Transformer backbone outperforms the convolutional backbone. We attribute this to the inherent noise present in ultrasound images. The Transformer's backbone is more adept at learning improved ultrasound image representations.
  • Incorporating all images during training is crucial. In the mmdetection dataloader configuration, the option filter_cfg=dict(filter_empty_gt=False) is utilized. This allows the model to see more negative examples during training.

Training the Model

To train our solution using the provided dino-5scale_swin-l_8xb2-12e_liver.py config file, follow these steps:

  1. Follow the instructions to install mmcv from the official documentation: mmcv Installation Guide
  2. Clone the mmdetection repository:
    git clone https://github.com/open-mmlab/mmdetection.git
    
  3. Download the COCO pretrained weights for SWIN-L from the provided link: SWIN-L Pretrained Weights
  4. Modify the paths in the provided config file dino-5scale_swin-l_8xb2-12e_liver.py to correspond to your dataset and pretrained weights location.
  5. Navigate to the mmdetection directory:
    cd mmdetection
    
  6. Begin training with:
    python tools/train.py path/to/dino-5scale_swin-l_8xb2-12e_liver.py
    

Model Configurations

For a detailed training configuration, refer to the file dino-5scale_swin-l_8xb2-12e_liver.py. Additional information on configuring models can be found in the mmdetection documentation.

Members

  • Zaw Htet Aung
  • Kittinan Srithaworn
  • Titipat Achakulvisut

liver-lesions-detection-2023's People

Contributors

z-zawhtet-a avatar titipata avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.