目录
GCN(Graph Convolutional Network,图卷积网络)
Curve-GCN是一种高效交互式图像标注方法,其性能优于Polygon-RNN++。在自动模式下运行时间为29.3ms,在交互模式下运行时间为2.6ms,比Polygon-RNN ++分别快10倍和100倍。
图像标注速度提升10倍!
多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!-CSDN博客
gcn推理入门例子:
import torch
from torch_geometric.nn import GCNConv
from torch_geometric.data import Data
class GCN(torch.nn.Module):
def __init__(self, num_features, num_classes):
super(GCN, self).__init__()
self.conv1