Neural network library for digit classification powered by cuda
The library was built to work with MNIST dataset. python-mnist data parser was used to load the dataset.
The library uses Numba which supports CUDA-enabled GPUs with compute capability 2.0 or above with an up-to-date Nvidia driver. See the list of CUDA-enabled GPU cards
Trained weights are located in weights directory. Call load_weights method to use them.
Py2CUDA Convert python code to CUDA. Usage To convert a python file say named py_file.py to CUDA, run python generate_cuda.py --file py_file.py --arch
Swin Transformer for Semantic Segmentation of satellite images This repo contains the supported code and configuration files to reproduce semantic seg
Zero-shot Natural Language Video Localization (ZSNLVL) by Pseudo-Supervised Video Localization (PSVL) This repository is for Zero-shot Natural Languag
Tracing Versus Freehand for Evaluating Computer-Generated Drawings (SIGGRAPH 2021) Zeyu Wang, Sherry Qiu, Nicole Feng, Holly Rushmeier, Leonard McMill
Static Features Classifier This is a static features classifier for Point-Could
Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)
Explicable Reward Design for Reinforcement Learning Agents [NeurIPS'21]
Face Attention Network Pytorch implementation of face attention network as described in Face Attention Network: An Effective Face Detector for the Occ
Memory-Efficient Multi-Level In-Situ Generation (MLG) By Jiaqi Gu, Hanqing Zhu, Chenghao Feng, Mingjie Liu, Zixuan Jiang, Ray T. Chen and David Z. Pan
InvariantAncestrySearch This repository contains python code necessary to replicated the experiments performed in our paper "Invariant Ancestry Search
Implementation of the ivis algorithm as described in the paper Structure-preserving visualisation of high dimensional single-cell datasets.
Random-walk Embedding Framework This repository is a reference implementation of the random-walk embedding framework as described in the paper: A Broa
PyTorch implementation of DUL (Data Uncertainty Learning in Face Recognition, CVPR2020)
Note: We're moving to PyTorch Lightning! Read about the move here. From the end of February, torchbearer will no longer be actively maintained. We'll
MosaicML Composer contains a library of methods, and ways to compose them together for more efficient ML training.
De(e)pendable Distributed Control of Port-Hamiltonian Systems (DeepDisCoPH) This repository is associated to the paper [1] and it contains: The full p
ChemRxiv | [Paper] XXX DeepStruc Welcome to DeepStruc, a Deep Generative Model (DGM) that learns the relation between PDF and atomic structure and the
YOLOv4-large This is the implementation of "Scaled-YOLOv4: Scaling Cross Stage Partial Network" using PyTorch framwork. YOLOv4-CSP YOLOv4-tiny YOLOv4-
Pano-AVQA Official repository of PanoAVQA: Grounded Audio-Visual Question Answering in 360° Videos (ICCV 2021) [Paper] [Poster] [Video] Getting Starte
TagLab: an image segmentation tool oriented to marine data analysis TagLab was created to support the activity of annotation and extraction of statist
3 Jul 14, 2021
23 Oct 10, 2022
37 Dec 27, 2022
23 Dec 09, 2022
1 Jan 25, 2022
2.4k Jan 08, 2023
3 May 12, 2022
312 Dec 09, 2022
2 Jan 04, 2022
0 Feb 02, 2022
285 Jan 04, 2023
23 Dec 13, 2022
20 Nov 15, 2022
631 Jan 04, 2023
2.8k Jan 08, 2023
3 Aug 17, 2022
13 Aug 01, 2022
2k Jan 02, 2023
9 Dec 23, 2022
49 Dec 29, 2022