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

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

This repo contains the pytorch implementation for Dynamic Concept Learner (accepted by ICLR 2021).

Implementation of "Meta-rPPG: Remote Heart Rate Estimation Using a Transductive Meta-Learner"

Code for "Long Range Probabilistic Forecasting in Time-Series using High Order Statistics"

Rocket-recycling with Reinforcement Learning

Code, Data and Demo for Paper: Controllable Generation from Pre-trained Language Models via Inverse Prompting

This repo contains the official code and pre-trained models for the Dynamic Vision Transformer (DVT).

Deep Learning Emotion decoding using EEG data from Autism individuals

Pytorch implementation of the paper "Topic Modeling Revisited: A Document Graph-based Neural Network Perspective"

Running AlphaFold2 (from ColabFold) in Azure Machine Learning

Implementation of "Fast and Flexible Temporal Point Processes with Triangular Maps" (Oral @ NeurIPS 2020)

FairMOT for Multi-Class MOT using YOLOX as Detector

DeepLab is a state-of-art deep learning system for semantic image segmentation built on top of Caffe.

Feature board for ERPNext

DeepStochlog Package For Python

Dynamic Visual Reasoning by Learning Differentiable Physics Models from Video and Language (NeurIPS 2021)

Code needed to reproduce the examples found in "The Temporal Robustness of Stochastic Signals"

Label Hallucination for Few-Shot Classification

Zero-shot Learning by Generating Task-specific Adapters

LSUN Dataset Documentation and Demo Code