Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR

🛠 Toolbox to extend PyTorch functionalities

Estimate/count FLOPS for a given neural network using pytorch

A small OpenCL benchmark program to measure peak GPU/CPU performance.

Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)

AI and Memory Wall

A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.

Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption

Count number of parameters / MACs / FLOPS for ONNX models.

Profile PyTorch models for FLOPs and parameters, helping to evaluate computational efficiency and memory usage.

FLOPs calculator with tf.profiler for neural network architecture written in tensorflow 2.2+ (tf.keras)

A toolkit for scaling law research ⚖

【一个小工具】用最短的代码完成对模型的分析，包含 ImageNet Val、FLOPs、Params、Throuthput、CAM 等

各种算法评价指标的实现（mAP/Flops/params/fps/error-rate/accuracy）

FLOPs and other statistics COunter for tf.keras neural networks

Discrete Flow Matching implemented in PyTorch

Computing FLOPs with Intel Software Development Emulator (Intel SDE)

FLOPs and other statistics COunter for Pytorch neural networks

CPU Benchmarking Tool which calculates IOPS and FLOPS.

Dynamic Frame Interpolation in Wavelet Domain (TIP 2023)