yet_another_pytorch_slot_attention:使用转换后的检查点在PyTorch中重新实现插槽注意(对象发现任务) 源码
Locatello等人(“对象发现”任务)的另一个的重新实现。 原始TFv2实现位于 培训循环代码改编自 ARI计算和CLEVRWithMasks代码改编自 # download original checkpoint (converted to PyTorch) wget https://github.com/vadimkantorov/yet_another_pytorch_slot_attention/releases/download/data/slot-attention_object_discovery.pt # infer on CPU with original checkpoint python infer.py --device cpu --checkpoint-tensorflow slot-attention_object_discovery.pt # down
文件列表
yet_another_pytorch_slot_attention-master.zip
(预估有个8文件)
yet_another_pytorch_slot_attention-master
infer.py
3KB
eval.py
2KB
models.py
8KB
README.md
2KB
metrics.py
1KB
clevr.py
2KB
clevr_with_masks.py
5KB
train.py
7KB
暂无评论