zhenshen-mla Goto Github PK

aesthetic-emotion-dataset

IAE Dataset, produced by Chaoran Cui, Zhen Shen, Jun Yu. A large scale dataset to facilitate multi-task learning for unified image aesthetics and emotion prediction.

aesthetic-semantic-joint-learning

afanet

Pytorch implementation of Paper："Adaptive Feature Aggregation in Deep Multi-task Convolutional Neural Networks."//Chaoran Cui, Zhen Shen, Jin Huang

allrank

allRank is a framework for training learning-to-rank neural models based on PyTorch.

ava_downloader

:arrow_double_down: Download AVA dataset (A Large-Scale Database for Aesthetic Visual Analysis)

captainblackboard

船长关于机器学习、计算机视觉和工程技术的总结和分享

cifar10-in-pytorch

continuousparetomtl

[ICML 2020] PyTorch Code for "Efficient Continuous Pareto Exploration in Multi-Task Learning"

depth-map-prediction-from-a-single-image-using-a-multi-scale-deep-network

PyTorch implementation of the paper

dsnet

DSNet: A Flexible Detect-to-Summarize Network for Video Summarization

faster-rcnn-pytorch

这是一个faster-rcnn的pytorch实现的库，可以利用voc数据集格式的数据进行训练。

fcrn

A Pytorch implement of 《Deeper Depth Prediction with Fully Convolutional Residual Networks》

grad-cam.pytorch

pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...

image-classification

We experiment on cifar-10 dataset using three kinds of network structures: VGG, ResNet and GoogLeNet.

luotuo-chinese-llm

骆驼(Luotuo): Open Sourced Chinese Language Models. Developed by 陈启源 @ 华中师范大学 & 李鲁鲁 @ 商汤科技 & 冷子昂 @ 商汤科技

machine-learning

:zap:机器学习实战（Python3）：kNN、决策树、贝叶斯、逻辑回归、SVM、线性回归、树回归

ml-in-action

The site is constantly updated. Machine Learning in Action!

multiobjectiveoptimization

Source code for Neural Information Processing Systems (NeurIPS) 2018 paper "Multi-Task Learning as Multi-Objective Optimization"

nddr-cnn-pytorch

PyTorch Implementation of "NDDR-CNN: Layerwise Feature Fusing in Multi-Task CNNs by Neural Discriminative Dimensionality Reduction"

paretomtl

Code for Neural Information Processing Systems (NeurIPS) 2019 paper: Pareto Multi-Task Learning

pcgrad

Code for "Gradient Surgery for Multi-Task Learning"

pytorch-deeplab-xception

DeepLab v3+ model in PyTorch. Support different backbones.

pytorch-grad-norm

Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning adjustable weight coefficients.

sam-detr

[CVPR'2022] SAM-DETR & SAM-DETR++: Official PyTorch Implementation

satellite-image-segmentation

A prototype system for semantic segmentation of Satellite Images. The executable file is 700MB, model parameters are 40MB, and the computation cost is 266 in GFLOPS.

series-photo-selection

Pytorch implementation of Paper："LEARNING MULTI-SCALE ATTENTIVE FEATURES FOR SERIES PHOTO SELECTION" //ICASSP2020//Jin Huang, Chaoran Cui, Chunyun Zhang, Zhen Shen

simple-faster-rcnn-pytorch

A simplified implemention of Faster R-CNN that replicate performance from origin paper

ssd-pytorch

这是一个ssd-pytorch的源码，可以用于训练自己的模型。

support-vector-machine

SVM有三宝，间隔、对偶、核技巧......

swin_trans