pytorchのスライス代入操作をonnxに変換する際にScatterNDならないようにするサンプル。
スライスしたtensorにそのまま代入してしまうとScatterNDになるため、計算結果をcatで新しいtensorにする。
Python 3.8.11
pip install -r requirements.txt
python convertonnx.py
scatternd_model.onnxとcat_model.onnxが作成されるのでNetronなどで確認。
ScatterNDになるような演算を含むモデル
ScatterNDを使わずscatterndmodelと同じ結果になるようにしたモデル
VoV3D is an efficient and effective 3D backbone network for temporal modeling implemented on top of PySlowFast. Diverse Temporal Aggregation and
English | 简体中文 Introduction MMCV is a foundational library for computer vision research and supports many research projects as below: MMCV: OpenMMLab
Run inference at scale Cortex is an open source platform for large-scale machine learning inference workloads. Workloads Realtime APIs - respond to pr
NeuralCompression is a Python repository dedicated to research of neural networks that compress data. The repository includes tools such as JAX-based entropy coders, image compression models, video c
Image De-raining Using a Conditional Generative Adversarial Network [Paper Link] [Project Page] He Zhang, Vishwanath Sindagi, Vishal M. Patel In this
High performance distributed framework for training deep learning recommendation models based on PyTorch.
Cobalt-Strike-det Cobalt Strike teamserver detection. usage: cobaltstrike_verify.py [-l TARGETS] [-t THREADS] optional arguments: -h, --help show this
Project looking into use of autoencoder for semi-supervised learning and comparing data requirements compared to supervised learning.
Variational Model Inversion Attacks Kuan-Chieh Wang, Yan Fu, Ke Li, Ashish Khisti, Richard Zemel, Alireza Makhzani Most commands are in run_scripts. W
To make the comparison with Animatable NeRF easier on the Human3.6M dataset, we save the quantitative results at here, which also contains the results of other methods, including Neural Body, D-NeRF,
Towards Implicit Text-Guided 3D Shape Generation Towards Implicit Text-Guided 3D Shape Generation (CVPR2022) Code for the paper [Towards Implicit Text
Python scripts for performing 3D human pose estimation using the Mobile Human Pose model in ONNX.
NeRViS: Neural Re-rendering for Full-frame Video Stabilization Project Page | Video | Paper | Google Colab Setup Setup environment for [Yu and Ramamoo
ResDAVEnet-VQ Official PyTorch implementation of Learning Hierarchical Discrete Linguistic Units from Visually-Grounded Speech What is in this repo? M
GainedVAE A Pytorch Implementation of a continuously rate adjustable learned image compression framework, Gained Variational Autoencoder(GainedVAE). N
Hyperparameter Optimization for Keras Talos • Key Features • Examples • Install • Support • Docs • Issues • License • Download Talos radically changes
Leveraging Instance-, Image- and Dataset-Level Information for Weakly Supervised Instance Segmentation This paper has been accepted and early accessed
A multi-scale unsupervised learning for deformable image registration Shuwei Shao, Zhongcai Pei, Weihai Chen, Wentao Zhu, Xingming Wu and Baochang Zha
TME The source codes for TME-BNA: Temporal Motif-Preserving Network Embedding with Bicomponent Neighbor Aggregation. Our implementation is based on TG
CRNN-PyTorch Implementation of https://arxiv.org/abs/1507.05717
102 Dec 06, 2022
4.6k Jan 09, 2023
7.9k Jan 06, 2023
297 Jan 06, 2023
216 Dec 18, 2022
340 Dec 30, 2022
17 Sep 27, 2022
2 Dec 17, 2021
15 Dec 26, 2022
359 Jan 08, 2023
55 Dec 16, 2022
99 Dec 31, 2022
9 Jun 17, 2022
21 Aug 23, 2022
39 Dec 24, 2022
1.6k Dec 15, 2022
39 Sep 20, 2022
2 Apr 13, 2022
2 Feb 10, 2022
4 Sep 01, 2022