PyTorch implementation of the U-Net for image semantic segmentation with high quality images

TPAMI：Frequency-aware Feature Fusion for Dense Image Prediction

Mamba SSM architecture

(CVPR2024 Highlight) Novel Class Discovery for Ultra-Fine-Grained Visual Categorization (UFG-NCD)

Collection of generative models in Pytorch version.

Image-to-Image Translation in PyTorch

Rice Yield CNN is a model to estimate the rice yield based on RGB image of rice canopy at harvest. The model is developed based on more than 22,000 images and yield database collected across 7 coun…

这是一个yolov8-pytorch的仓库，可以用于训练自己的数据集。

Open source Farm management Information System (FMIS) - Connecting farms to the world

[CVPRW 2023] SCANet: Self-Paced Semi-Curricular Attention Network for Non-Homogeneous Image Dehazing

A unified 3D Transformer Pipeline for visual synthesis

The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

MetaSeg: Packaged version of the Segment Anything repository

这是一个unet-pytorch的源码，可以训练自己的模型

M2DGR： a Multi-modal and Multi-scenario Dataset for Ground Robots(RA-L2021 & ICRA2022)

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

🔥 Mask R-CNN and Keypoint R-CNN api wrapper in PyTorch.

Number detection implemented using TensorFlow with custom CNN architecture for fast inference and custom dataset