Effect of Deep Transfer and Multi task Learning on Sperm Abnormality Detection

Last update: Sep 05, 2022

Overview

Effect of Deep Transfer and Multi task Learning on Sperm Abnormality Detection

Introduction

This repository includes codes and models of "Effect of Deep Transfer and Multi task Learning on Sperm Abnormality Detection" paper. link: https://doi.org/10.1016/j.compbiomed.2020.104121

Dataset

First you should download the MHSMA dataset using:

git clone https://github.com/soroushj/mhsma-dataset.git

Usage

First of all,the configuration file should be setted.So open dmtl.txt or dtl.txt and set the setting you want.This files contains paramaters of the model that you are going to train.

dtl.txt have only one line and contains paramaters to train a DTL model.
dmtl.txt contains two lines:paramaters of stage 1 are kept in the first line of the file and paramaters of stage 2 are kept in the second line of the file.
Some paramaters have an aray of three values that they keep the value of three labels.To set them,consider this sequence:[Acrosome,Vacoule,Head].
To train a DTL model,use the following commands and arguments:

python train.py -t dtl [-e epchos] [-label label]  [-model model] [-w file]

Argumetns:

Argument	Description
-t	type of network(dtl or dmtl)
-e	number of epochs
-label	label(a,v or h)
-model	pre-trained model
-w	name of best weihgt file
--phase	You can use it to choose stage in DMTL(1 or 2)
--second_model	The base model for second stage of DMTL

1.Train

To choose a pre-trained model, you can use one of the following models:

model argument	Description
vgg_19	VGG 19
vgg_16	VGG 16
resnet_50	Resnet 50
resnet_101	Resnet 101
resnet_502	Resnet 502

To train a DMTL model,use the following commands and arguments:

python train.py -t dmtl [--phase phase] [-e epchos] [-label label] [-model model] [-w file]

Also you can use your own pre-trained model by using address of your model instead of the paramaters been told in the table above.

Example:
python train.py -t dmtl --phase 1 -e 100 -label a -model C:\model.h5 -w w.h5

2.K Fold

To perform K Fold on a model,use "-k_fold True" argument.

python train.py -k_fold True [-t type] [-e epchos] [-label label] [-model model] [-w file]

3.Threshold Search

To find a good threshold for your model,use the following code:

python threshold.py [-t type] [-addr model address] [-l label]

Models

The CNN models that were introduced and evaluated in our research paper can be found in the v1.0 release of this repository.

You might also like...

Face Detection and Alignment using Multi-task Cascaded Convolutional Networks (MTCNN)

Face-Detection-with-MTCNN Face detection is a computer vision problem that involves finding faces in photos. It is a trivial problem for humans to sol

3 Oct 7, 2022

Multi-task yolov5 with detection and segmentation based on yolov5

YOLOv5DS Multi-task yolov5 with detection and segmentation based on yolov5(branch v6.0) decoupled head anchor free segmentation head README中文 Ablation

150 Dec 30, 2022

Code for the ICML 2021 paper "Bridging Multi-Task Learning and Meta-Learning: Towards Efficient Training and Effective Adaptation", Haoxiang Wang, Han Zhao, Bo Li.

Bridging Multi-Task Learning and Meta-Learning Code for the ICML 2021 paper "Bridging Multi-Task Learning and Meta-Learning: Towards Efficient Trainin

57 Dec 15, 2022

A novel Engagement Detection with Multi-Task Training (ED-MTT) system

Comments

a possible typo(bug)

Very interesting idea and complements!

In LoadData.py, starting from line 150, ` if phase == 'search':

    return {
            "x_train": x_train_128,
            "y_train": y_train,
            "x_train_128": x_train_128,
            'x_val_128': x_valid_128,
            "x_val": x_valid_128,
            "y_val": y_valid,
            "x_test": x_test_128,
            "y_test": y_test
            }`

here, I think that the first key-value pair should probably be "x_train": x_train instead of "x_train": x_train_128, which causes an error of shape mismatch during fit.

opened by captainst 0

Effect of Deep Transfer and Multi task Learning on Sperm Abnormality Detection

Related tags

Overview

Effect of Deep Transfer and Multi task Learning on Sperm Abnormality Detection

Introduction

Dataset

Usage

1.Train

2.K Fold

3.Threshold Search

Models

You might also like...

Face Detection and Alignment using Multi-task Cascaded Convolutional Networks (MTCNN)

Multi-task yolov5 with detection and segmentation based on yolov5

Code for the ICML 2021 paper "Bridging Multi-Task Learning and Meta-Learning: Towards Efficient Training and Effective Adaptation", Haoxiang Wang, Han Zhao, Bo Li.

A novel Engagement Detection with Multi-Task Training (ED-MTT) system

Self-training for Few-shot Transfer Across Extreme Task Differences

Effect of Different Encodings and Distance Functions on Quantum Instance-based Classifiers

Efficient neural networks for analog audio effect modeling

[CVPR 2021] Counterfactual VQA: A Cause-Effect Look at Language Bias

Algebraic effect handlers in Python

Comments

a possible typo(bug)

Releases(v1.0)

v1.0(Dec 29, 2020)

Owner

Amir Abbasi

A High-Quality Real Time Upscaler for Anime Video

Machine Learning toolbox for Humans

Official implementation of "Implicit Neural Representations with Periodic Activation Functions"

Official PyTorch implementation of Less is More: Pay Less Attention in Vision Transformers.

PyTorch implementation for STIN

DyNet: The Dynamic Neural Network Toolkit

Using the provided dataset which includes various book features, in order to predict the price of books, using various proposed methods and models.

Unrolled Variational Bayesian Algorithm for Image Blind Deconvolution

Audio Source Separation is the process of separating a mixture into isolated sounds from individual sources

House3D: A Rich and Realistic 3D Environment

Code for the paper: Hierarchical Reinforcement Learning With Timed Subgoals, published at NeurIPS 2021

naked is a Python tool which allows you to strip a model and only keep what matters for making predictions.

Source code for Transformer-based Multi-task Learning for Disaster Tweet Categorisation (UCD's participation in TREC-IS 2020A, 2020B and 2021A).

Code for Blind Image Decomposition (BID) and Blind Image Decomposition network (BIDeN).

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.

TuckER: Tensor Factorization for Knowledge Graph Completion

A collection of differentiable SVD methods and also the official implementation of the ICCV21 paper "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance Pooling?"

Official implementation for "Symbolic Learning to Optimize: Towards Interpretability and Scalability"

[CVPR 2016] Unsupervised Feature Learning by Image Inpainting using GANs

TensorLight - A high-level framework for TensorFlow