Shuffle Attention for MobileNetV3
Run the following command for train model on your own dataset:
python train.py --dataset mnist
Run the following command for evaluation trained model on test dataset:
python test.py --dataset mnist
Run the following command for classification images:
python inference.py --input /path/to/image.jpg
Introduction This repository is for X-Linear Attention Networks for Image Captioning (CVPR 2020). The original paper can be found here. Please cite wi
PyTorch code for the paper "Curriculum Graph Co-Teaching for Multi-target Domain Adaptation" (CVPR2021) This repo presents PyTorch implementation of M
Adversarial vulnerability of powerful near out-of-distribution detection by Stanislav Fort In this repository we're collecting replications for the ke
A python package simulating the quasi-2D pseudospin-1/2 Gross-Pitaevskii equation with NVIDIA GPU acceleration. Introduction spinor-gpe is high-level,
USSBriefs2021 This provides the R code and data to replicate results in "The USS Trustee’s risky strategy" by Neil M Davies, Jackie Grant and Chin Yan
CS_Final_Metal_surface_detection This is a final project for CoderSchool Machine Learning bootcamp on 29/12/2021. The project is based on the dataset
Detecting Blurred Ground-based Sky/Cloud Images With the spirit of reproducible research, this repository contains all the codes required to produce t
CM-NAS Official Pytorch code of paper CM-NAS: Cross-Modality Neural Architecture Search for Visible-Infrared Person Re-Identification in ICCV2021. Vis
Localization Distillation for Object Detection This repo is based on mmDetection. This is the code for our paper: Localization Distillation
Self-Classifier: Self-Supervised Classification Network Official PyTorch implementation and pretrained models of the paper Self-Supervised Classificat
GPyTorch GPyTorch is a Gaussian process library implemented using PyTorch. GPyTorch is designed for creating scalable, flexible, and modular Gaussian
Rotary Embeddings - Pytorch A standalone library for adding rotary embeddings to transformers in Pytorch, following its success as relative positional
Granger causality discovery for neural networks.
Semi-Supervised Domain Generalization This code is the official implementation of the following paper: Semi-Supervised Domain Generalization with Stoc
Pytorch Negative Sampling Loss Negative Sampling Loss implemented in PyTorch. Usage neg_loss = NEG_loss(num_classes, embedding_size) optimizer =
image-processing This module generates depth maps for shape-blocked-out images Install If working with anaconda, then from the root directory: conda e
Credit Card Fraud Detection Came across this mocked-up dataset of customer transactions at [Capital One Recruitment Challenge](https://github.com/Capi
Status: Maintenance (expect bug fixes and minor updates) OpenAI Gym OpenAI Gym is a toolkit for developing and comparing reinforcement learning algori
DIGAN (ICLR 2022) Official PyTorch implementation of "Generating Videos with Dyn
ManiSkill Benchmark SAPIEN Manipulation Skill Benchmark (abbreviated as ManiSkill, pronounced as "Many Skill") is a large-scale learning-from-demonstr
240 Dec 17, 2022
79 Dec 19, 2022
9 Aug 30, 2022
2 Sep 20, 2022
1 Oct 30, 2021
1 Dec 29, 2021
1 Oct 20, 2021
274 Dec 26, 2022
24 Dec 21, 2022
3k Jan 02, 2023
110 Dec 30, 2022
130 Jan 01, 2023
49 Dec 10, 2022
123 Sep 13, 2022
2 Oct 17, 2022
1 Nov 17, 2022
29.6k Jan 08, 2023
147 Dec 31, 2022
107 Jan 08, 2023