Official code of CVPR 2021's PLOP: Learning without Forgetting for Continual Semantic Segmentation

Overview

PLOP: Learning without Forgetting for Continual Semantic Segmentation

Paper Conference

Vizualization on VOC 15-1

This repository contains all of our code. It is a modified version of Cermelli et al.'s repository.

@inproceedings{douillard2021plop,
  title={PLOP: Learning without Forgetting for Continual Semantic Segmentation},
  authors={Douillard, Arthur and Chen, Yifu and Dapogny, Arnaud and Cord, Matthieu},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
  year={2021}
}

Requirements

You need to install the following libraries:

  • Python (3.6)
  • Pytorch (1.7.1)
  • torchvision (0.4.0)
  • tensorboardX (1.8)
  • apex (0.1)
  • matplotlib (3.3.1)
  • numpy (1.17.2)
  • inplace-abn (1.0.7)

Note also that apex seems to only work with some CUDA versions, therefore try to install Pytorch with the 9.2 or 10.0 CUDA versions, do:

conda install -y pytorch torchvision cudatoolkit=9.2 -c pytorch
cd apex
pip3 install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

Dataset

Two scripts are available to download ADE20k and Pascal-VOC 2012, please see in the data folder. For Cityscapes, you need to do it yourself, because you have to ask "permission" to the holders; but be reassured, it's only a formality, you can get the link in a few days by mail.

Performance on VOC

How to perform training

The most important file is run.py, that is in charge to start the training or test procedure. To run it, simpy use the following command:

python -m torch.distributed.launch --nproc_per_node=<num_GPUs> run.py --data_root <data_folder> --name <exp_name> .. other args ..

The default is to use a pretraining for the backbone used, that is searched in the pretrained folder of the project. We used the pretrained model released by the authors of In-place ABN (as said in the paper), that can be found here: link. I've also upload those weights there: link.

Since the pretrained are made on multiple-gpus, they contain a prefix "module." in each key of the network. Please, be sure to remove them to be compatible with this code (simply rename them using key = key[7:]) (if you're working on single gpu). If you don't want to use pretrained, please use --no-pretrained.

There are many options (you can see them all by using --help option), but we arranged the code to being straightforward to test the reported methods. Leaving all the default parameters, you can replicate the experiments by setting the following options.

  • please specify the data folder using: --data_root <data_root>
  • dataset: --dataset voc (Pascal-VOC 2012) | ade (ADE20K)
  • task: --task <task>, where tasks are
    • 15-5, 15-5s, 19-1 (VOC), 100-50, 100-10, 50, 100-50b, 100-10b, 50b (ADE, b indicates the order)
  • step (each step is run separately): --step <N>, where N is the step number, starting from 0
  • (only for Pascal-VOC) disjoint is default setup, to enable overlapped: --overlapped
  • learning rate: --lr 0.01 (for step 0) | 0.001 (for step > 0)
  • batch size: --batch_size <24/num_GPUs>
  • epochs: --epochs 30 (Pascal-VOC 2012) | 60 (ADE20K)
  • method: --method <method name>, where names are
    • FT, LWF, LWF-MC, ILT, EWC, RW, PI, MIB

For all details please follow the information provided using the help option.

Example commands

LwF on the 100-50 setting of ADE20K, step 0:

python -m torch.distributed.launch --nproc_per_node=2 run.py --data_root data --batch_size 12 --dataset ade --name LWF --task 100-50 --step 0 --lr 0.01 --epochs 60 --method LWF

MIB on the 50b setting of ADE20K, step 2:

python -m torch.distributed.launch --nproc_per_node=2 run.py --data_root data --batch_size 12 --dataset ade --name MIB --task 100-50 --step 2 --lr 0.001 --epochs 60 --method MIB

LWF-MC on 15-5 disjoint setting of VOC, step 1:

python -m torch.distributed.launch --nproc_per_node=2 run.py --data_root data --batch_size 12 --dataset voc --name LWF-MC --task 15-5 --step 1 --lr 0.001 --epochs 30 --method LWF-MC

PLOP on 15-1 overlapped setting of VOC, step 1:

python -m torch.distributed.launch --nproc_per_node=2 run.py --data_root data --batch_size 12 --dataset voc --name PLOP --task 15-5s --overlapped --step 1 --lr 0.001 --epochs 30 --method FT --pod local --pod_factor 0.01 --pod_logits --pseudo entropy --threshold 0.001 --classif_adaptive_factor --init_balanced --pod_options "{"switch": {"after": {"extra_channels": "sum", "factor": 0.0005, "type": "local"}}}"

Once you trained the model, you can see the result on tensorboard (we perform the test after the whole training) or you can test it by using the same script and parameters but using the command

--test

that will skip all the training procedure and test the model on test data.

Or more simply you can use one of the provided script that will launch every step of a continual training.

For example, do

bash scripts/plop_15-1.sh

Note that you will need to modify those scripts to include the path where your data.

Comments
  • Extremely hard incremental scenario

    Extremely hard incremental scenario

    Thank you for your great work, PLOP!

    As your code, we can reproduce the performance almost the same as your paper.

    Additionally, we also conduct experiments on extremely hard incremental scenarios, such as 5-1 (16 steps) and 2-1 (19 steps).

    for this, we add below lines on task.py

    "5-1": {
            0 : [0, 1, 2, 3, 4, 5],
            1 : [6, ],
            2 : [7, ],
            3 : [8, ],
            4 : [9, ],
            5 : [10, ],
            6 : [11, ],
            7 : [12, ],
            8 : [13, ],
            9 : [14, ],
            10: [15, ],
            11: [16, ],
            12: [17, ],
            13: [18, ],
            14: [19, ],
            15: [20, ],
        },   
        "2-1":{
            0 : [0, 1, 2],
            1 : [3, ],
            2 : [4, ],
            3 : [5, ],
            4 : [6, ],
            5 : [7, ],
            6 : [8, ],
            7 : [9, ],
            8 : [10, ],
            9 : [11, ],
            10: [12, ],
            11: [13, ],
            12: [14, ],
            13: [15, ],
            14: [16, ],
            15: [17, ],
            16: [18, ],
            17: [19, ],
            18: [20, ],
        },   
    

    However, during the training, the loss is divergence to nan.

    I already noticed that someone suffers from the loss divergence issue https://github.com/arthurdouillard/CVPR2021_PLOP/issues/8 on 15-5 task, however, I can reproduce the performance on 15-5 task in my environmental settings. Also, MiB on these extremely hard scenarios was well trained without the loss divergence, however, PLOP showed the issue.

    Therefore, I wonder you also have the same issue in the extremely hard scenarios, 5-1 and 2-1. And, please tell me how can I solve that issue (e.g., which hyperparameter should be changed).

    Thanks.

    opened by qjadud1994 11
  • Model weights become NaN in step 1 on VOC

    Model weights become NaN in step 1 on VOC

    Hi, Thanks for your contribution. I have a problem when training PLOP on VOC dataset with setting 15-5. After successfully training the model at step 0, I trained the model at the next step. The model becomes NaN after few training iterations even at the first epoch.

    Since the model M0 can be trained without any problem, I doubt that the distilling the knowledge of M0 to M1 might lead to a divergence problem for M1.

    Following the papers, I used lr=0.01 for M0 and lr=0.001 for M1. Here is the setting I used. Step 0: python -m torch.distributed.launch --nproc_per_node=2 run.py --data_root /media/hieu/DATA/semantic_segmentation/PascalVOC12 --batch_size 12 --dataset voc --name PLOP --task 15-5 --overlap --step 0 --lr 0.01 --epochs 30 --method FT --pod local --pod_factor 0.01 --pod_logits --pseudo entropy --threshold 0.001 --classif_adaptive_factor --init_balanced

    Step 1: python -m torch.distributed.launch --nproc_per_node=2 run.py --data_root /media/hieu/DATA/semantic_segmentation/PascalVOC12 --batch_size 12 --dataset voc --name PLOP --task 15-5 --overlap --step 1 --lr 0.001 --epochs 30 --method FT --pod local --pod_factor 0.01 --pod_logits --pseudo entropy --threshold 0.001 --classif_adaptive_factor --init_balanced.

    opened by HieuPhan33 10
  • Loss is not converge

    Loss is not converge

    While I am reimplementing your code with your setting given in the scripts folder, I found the results are a bit lower than the paper results(2%-5%). When I check the tensorboard for the loss, I found that from step 1, the loss is not converging and some of them are NaN. image

    Have you ever run into this problem?

    opened by fred206968 7
  • Clarification regarding domain shift experiments on Cityscapes

    Clarification regarding domain shift experiments on Cityscapes

    Hi @arthurdouillard, I really enjoyed reading your work! Thanks for bringing in the domain shift aspect of CSS. I have the following doubts in the implementation of ILT, MiB and PLOP for the domain shift experiments on Cityscapes (Table 5):

    1. wrt PLOP: I'm assuming the pseudo labeling will not be applicable in these experiments as the label spaces are fixed in the domain incremental scenario. So do I just use the distillation loss along with regular cross-entropy? Is my understanding correct wrt using PLOP in a domain IL scenario?
    2. MiB modifies distillation and cross-entropy to tackle the background class shift issue. Since there is no such issue in the domain incremental scenario, doesn't their method get reduced to ILT (basically LwF)? I'm confused as to why there is a difference in the performance (For e.g. 59% for ILT and 61.5% for MiB in the 11-5 case).

    Also, is it possible to share the joint model (traditional segmentation model) mIoU you get for Cityscapes on DeeplabV3, ResNet101? (I couldn't find this in the paper and wanted to see the drop wrt the joint one).

    opened by prachigarg23 6
  • can't reproduce Ade20k and don't know the detail of Cityscapes.

    can't reproduce Ade20k and don't know the detail of Cityscapes.

    Hi, sorry to disturb but i have a few questions as followed.

    1. Maybe you have changed some details about the experiments over Cityscapes, since it seems like MiB cannot work on the Cityscapes because there are only GT without pseudo labels. I'd appreciate it if you can tell me what the exact details you have changed over these experiments.
    2. I cannot reproduce the performance that you mentioned in your paper about the Ade20k, having seen that you had reproduced the performance of your method over Cityscapes in another issue, I hope that you can reproduce it over Ade20k too and have a further discussion with me. Wish you good luck.
    opened by schuy1er 6
  • Reproduce 15-1 setup on Pascal VOC

    Reproduce 15-1 setup on Pascal VOC

    Describe the bug I tried to run the provided pascal VOC script using Apex optimization 01 and everything same as script except i was using a single GPU and hence changed the batch size to 24. But I got the following results

    | | 1-15 | 16-20 | all | |-------------|-------|-------|-------| | Paper | 65.12 | 21.11 | 54.64 | | Code results | 58.73 | 21.6 | 49.7 |

    To Reproduce start=date +%s`

    START_DATE=$(date '+%Y-%m-%d')

    PORT=$((9000 + RANDOM % 1000)) GPU=0 NB_GPU=1 DATA_ROOT=./data DATASET=voc TASK=15-5s NAME=PLOP METHOD=PLOP BATCH_SIZE=24 INITIAL_EPOCHS=30 EPOCHS=30 OPTIONS="--checkpoint checkpoints/step/"

    RESULTSFILE=results/${START_DATE}${DATASET}${TASK}_${NAME}.csv rm -f ${RESULTSFILE}

    CUDA_VISIBLE_DEVICES=${GPU} python3 -m torch.distributed.launch --master_port ${PORT} --nproc_per_node=${NB_GPU} run.py --date ${START_DATE} --data_root ${DATA_ROOT} --overlap --batch_size ${BATCH_SIZE} --dataset ${DATASET} --name ${NAME} --task ${TASK} --step 0 --lr 0.01 --epochs ${INITIAL_EPOCHS} --method ${METHOD} --opt_level O1 ${OPTIONS} for step in 1 2 3 4 5 do CUDA_VISIBLE_DEVICES=${GPU} python3 -m torch.distributed.launch --master_port ${PORT} --nproc_per_node=${NB_GPU} run.py --date ${START_DATE} --data_root ${DATA_ROOT} --overlap --batch_size ${BATCH_SIZE} --dataset ${DATASET} --name ${NAME} --task ${TASK} --step ${step} --lr 0.001 --epochs ${EPOCHS} --method ${METHOD} --opt_level O1 ${OPTIONS} done python3 average_csv.py ${RESULTSFILE}`

    opened by mostafaelaraby 5
  • Reproducing the experiments on Cityscapes

    Reproducing the experiments on Cityscapes

    We cannot reproduce the performance on Cityscapes. What are your training details? We get the last mIoU is 49.73 on 11-1 task. We set the initial learning rate as 0.01 and run 60 epochs. And we set the later learning rate as 0.001 and run 60 epochs. The batch size is set to 24.

    opened by zhangchbin 5
  • About apex=0.1

    About apex=0.1

    I config the same pytorch, torchvision and cuda version with you, however, I can not install apex=0.1. Can you tell you how to install it in detail? Thank you!

    opened by congwei45 4
  • predict and visualization

    predict and visualization

    Describe the bug Hi , I have one question about how to predict a picture when finish the training stage, in other words, how to visualize the segmentation results in your code. I have written a simple visualization .py file, but when I load the model , I found that the model did not load successfully. If it's convenient, I would appreciate it if you could share your predict code. We look forward to your answer, thanks

    To Reproduce

    Dataset: voc-2012 Setting: 19-1 Command used or script used:

    Expected behavior A clear and concise description of what you expected to happen.

    Additional context Add any other context about the problem here.

    opened by Liuhao-128 4
  • Cityscapes Dataset Preparation

    Cityscapes Dataset Preparation

    I'm struggling to reproduce results on CityScapes dataset like others already reported the issue. I found that in your paper you mentioned that with domain-incremental setting on CityScapes datasets you have used images from 21 cities in a particular order. However, after checking the dataset downloaded from www.cityscapes-dataset.com, i only can se 18 cities in train directory.

    Now my question is as followes:

    1. Where do you get images from those three extra cities, i.e. frankfurt, lindau, and munster.
    2. Could you please share more details on preparing CityScapes dataset

    Thanks for your time.

    opened by bit0123 3
  • Paper reproduction

    Paper reproduction

    Hello, I have run your code model on my server. I know your code has completed the training and prediction, but I can't find the prediction picture in the file. Only the predicted score, my deep learning ability is limited, so I ask you for help on how to get the segmented image in the paper. Thank you. My email: [email protected]

    opened by AI-student-wl 3
  • About visualiztion on ADE20K

    About visualiztion on ADE20K

    Describe the bug Hi, I have run the code on 50-50-50 setting on ade20k. I want to know how to visualize on the verification set. If you have the corresponding code, can you provide it?

    Thank you for your work and look forward to your reply

    To Reproduce

    Dataset: ADE20K Setting: 50-50-50

    opened by Liuhao-128 0
  • Trying to reproduce cityscapes domain results

    Trying to reproduce cityscapes domain results

    Describe the bug A clear and concise description of what the bug is.

    To Reproduce

    Dataset: City-scapes domain Setting: 1-1 task Command used or script used: I cloned the github repository and I was able to reproduce the results for FT(fine-tuning) method using the scripts. When I tried using PLOP method, after the task 0, I am getting the following error. image

    opened by anouskashrestha 0
  • Try to reproduce voc 10-1 results

    Try to reproduce voc 10-1 results

    I am trying to reproduce 10-1 results as shown in the table below. I notice a large gap of the old class mIoU between my reproduce result (38.82) and your reported one (44.03), roughly 5 percent points. I am wondering what will cause this problem. I run the experiments with 2 x RTX 3090 GPU. I follow your original implementation except for the cuda version. I am using cuda 11.3 because cuda 10.2 does not support RTX 3090. Does it matter?

    Btw, may I know what GPU model do you use? I think it requires to have at least 16G to hold a batch of 12 on each device and needs to support cuda 10.2 as well. V100? I guess.

    Meanwhile, I notice a weird phenomenon that background performance drops drastically starting from the 8-th step and becomes 0 at 9-th step. I think this harms the old class performance a lot. Do you have a similar issue?

    Thanks.

    | step | background | aeroplane | bicycle | bird | boat | bottle | bus | car | cat | chair | cow | diningtable | dog | horse | motorbike | person | pottedplant | sheep | sofa | train | tvmonitor | 0-10 | 11-20 | all | |:----:|:----------:|:---------:|:-------:|:------:|:------:|:------:|:------:|:------:|:------:|:------:|:------:|:-----------:|:------:|:------:|:---------:|:------:|:-----------:|:------:|:-----:|:-----:|:---------:|:------:|:------:|:------:| | 0 | 95.19% | 89.84% | 41.01% | 89.74% | 72.24% | 85.02% | 95.29% | 88.26% | 93.30% | 43.15% | 92.25% | x | x | x | x | x | x | x | x | x | x | 80.48% | - | 80.48% | | 1 | 92.01% | 87.25% | 35.76% | 82.88% | 67.51% | 79.44% | 94.49% | 82.55% | 88.44% | 46.25% | 90.74% | 37.38% | x | x | x | x | x | x | x | x | x | 77.03% | 37.38% | 73.73% | | 2 | 89.16% | 84.00% | 34.32% | 83.28% | 63.44% | 68.45% | 92.90% | 82.54% | 82.37% | 31.68% | 84.48% | 24.68% | 62.59% | x | x | x | x | x | x | x | x | 72.42% | 43.64% | 67.99% | | 3 | 83.97% | 79.07% | 31.92% | 73.50% | 53.19% | 51.06% | 91.18% | 80.90% | 76.82% | 17.40% | 65.26% | 18.69% | 3.93% | 22.11% | x | x | x | x | x | x | x | 64.02% | 14.91% | 53.50% | | 4 | 79.89% | 77.61% | 35.83% | 74.77% | 47.67% | 49.70% | 90.73% | 78.02% | 79.41% | 13.21% | 65.81% | 17.77% | 9.43% | 20.52% | 33.81% | x | x | x | x | x | x | 62.97% | 20.38% | 51.61% | | 5 | 84.56% | 84.34% | 37.88% | 81.72% | 53.81% | 56.75% | 90.72% | 82.48% | 82.98% | 13.77% | 69.20% | 0.57% | 0.00% | 26.10% | 58.37% | 69.71% | x | x | x | x | x | 67.11% | 30.95% | 55.81% | | 6 | 82.18% | 80.07% | 37.93% | 69.65% | 49.38% | 55.32% | 89.74% | 81.63% | 77.15% | 14.10% | 61.90% | 0.48% | 0.00% | 23.42% | 51.28% | 64.77% | 5.04% | x | x | x | x | 63.55% | 24.16% | 49.65% | | 7 | 80.42% | 77.61% | 36.88% | 50.15% | 42.00% | 53.55% | 75.10% | 80.31% | 74.55% | 10.30% | 20.94% | 0.05% | 0.00% | 24.42% | 50.90% | 57.63% | 0.00% | 19.33% | x | x | x | 54.71% | 21.76% | 41.90% | | 8 | 44.45% | 71.53% | 35.75% | 51.99% | 44.17% | 50.20% | 80.98% | 76.45% | 69.19% | 29.32% | 40.46% | 0.18% | 0.00% | 22.80% | 51.03% | 71.25% | 0.00% | 15.88% | 3.26% | x | x | 54.04% | 20.55% | 39.94% | | 9 | 0.55% | 61.80% | 35.53% | 51.15% | 44.62% | 50.52% | 69.98% | 77.75% | 57.82% | 9.92% | 16.76% | 0.05% | 0.00% | 23.56% | 46.89% | 62.95% | 0.00% | 7.37% | 1.49% | 2.27% | x | 43.31% | 16.06% | 31.05% | | 10 | 0.00% | 50.48% | 32.40% | 38.14% | 40.64% | 51.90% | 62.86% | 69.76% | 56.53% | 17.47% | 6.83% | 0.02% | 0.00% | 23.25% | 53.95% | 66.92% | 0.00% | 1.73% | 1.67% | 0.04% | 2.60% | 38.82% | 15.02% | 27.49% |

    opened by Ze-Yang 3
  • ade 100-10 reproduce

    ade 100-10 reproduce

    i used the script you gave and i only got 24.xx final mIoU, i don't know if it is related to specific seed?

    by the way, due to the limitation of gpu memory, i used 4 gpus with 4 batch each gpu, which was equal to total batch size of 24, and the bn is iabc_sync, i thought that 4 gpus is not a issue?

    any idea how i can reproduce your ade 100-10 results?

    opened by zhaoedf 3
  • The weight file is corrupted

    The weight file is corrupted

    I click the link and it shows that the weight file is damaged. What is the reason? Can I upload it again? I am working on my graduation project and looking forward to your reply. Thank you! Expected behavior A clear and concise description of what you expected to happen.

    Additional context Add any other context about the problem here.

    opened by gjiee 0
Releases(v1.01)
Owner
Arthur Douillard
PhD Student @ Sorbonne, Research Scientist @ Heuritech, Lecturer @ EPITA
Arthur Douillard
Official PyTorch Implementation of Unsupervised Learning of Scene Flow Estimation Fusing with Local Rigidity

UnRigidFlow This is the official PyTorch implementation of UnRigidFlow (IJCAI2019). Here are two sample results (~10MB gif for each) of our unsupervis

Liang Liu 28 Nov 16, 2022
This is a re-implementation of TransGAN: Two Pure Transformers Can Make One Strong GAN (CVPR 2021) in PyTorch.

TransGAN: Two Transformers Can Make One Strong GAN [YouTube Video] Paper Authors: Yifan Jiang, Shiyu Chang, Zhangyang Wang CVPR 2021 This is re-implem

Ahmet Sarigun 79 Jan 05, 2023
It is an open dataset for object detection in remote sensing images.

RSOD-Dataset It is an open dataset for object detection in remote sensing images. The dataset includes aircraft, oiltank, playground and overpass. The

136 Dec 08, 2022
Source code of CIKM2021 Long Paper "PSSL: Self-supervised Learning for Personalized Search with Contrastive Sampling".

PSSL Source code of CIKM2021 Long Paper "PSSL: Self-supervised Learning for Personalized Search with Contrastive Sampling". It consists of the pre-tra

2 Dec 21, 2021
Repository for training material for the 2022 SDSC HPC/CI User Training Course

hpc-training-2022 Repository for training material for the 2022 SDSC HPC/CI Training Series HPC/CI Training Series home https://www.sdsc.edu/event_ite

sdsc-hpc-training-org 21 Jul 27, 2022
Sign Language Translation with Transformers (COLING'2020, ECCV'20 SLRTP Workshop)

transformer-slt This repository gathers data and code supporting the experiments in the paper Better Sign Language Translation with STMC-Transformer.

Kayo Yin 107 Dec 27, 2022
Text-to-SQL in the Wild: A Naturally-Occurring Dataset Based on Stack Exchange Data

SEDE SEDE (Stack Exchange Data Explorer) is new dataset for Text-to-SQL tasks with more than 12,000 SQL queries and their natural language description

Rupert. 83 Nov 11, 2022
(NeurIPS 2020) Wasserstein Distances for Stereo Disparity Estimation

Wasserstein Distances for Stereo Disparity Estimation Accepted in NeurIPS 2020 as Spotlight. [Project Page] Wasserstein Distances for Stereo Disparity

Divyansh Garg 92 Dec 12, 2022
Deep Learning and Reinforcement Learning Library for Scientists and Engineers 🔥

TensorLayer is a novel TensorFlow-based deep learning and reinforcement learning library designed for researchers and engineers. It provides an extens

TensorLayer Community 7.1k Dec 29, 2022
fklearn: Functional Machine Learning

fklearn: Functional Machine Learning fklearn uses functional programming principles to make it easier to solve real problems with Machine Learning. Th

nubank 1.4k Dec 07, 2022
Label-Free Model Evaluation with Semi-Structured Dataset Representations

Label-Free Model Evaluation with Semi-Structured Dataset Representations Prerequisites This code uses the following libraries Python 3.7 NumPy PyTorch

8 Oct 06, 2022
The official PyTorch code for NeurIPS 2021 ML4AD Paper, "Does Thermal data make the detection systems more reliable?"

MultiModal-Collaborative (MMC) Learning Framework for integrating RGB and Thermal spectral modalities This is the official code for NeurIPS 2021 Machi

NeurAI 12 Nov 02, 2022
Trainable Bilateral Filter Layer (PyTorch)

Trainable Bilateral Filter Layer (PyTorch) This repository contains our GPU-accelerated trainable bilateral filter layer (three spatial and one range

FabianWagner 26 Dec 25, 2022
Detecting Potentially Harmful and Protective Suicide-related Content on Twitter

TwitterSuicideML Scripts for reproducing the Machine Learning analysis of the paper: Detecting Potentially Harmful and Protective Suicide-related Cont

3 Oct 17, 2022
Face Recognition Attendance Project

Face-Recognition-Attendance-Project In This Project You will learn how to mark attendance using face recognition, Hello Guys This is Gautam Kumar, Thi

Gautam Kumar 1 Dec 03, 2022
Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecasting

Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecasting This is the origin Pytorch implementation of Informer in the followin

Haoyi 3.1k Dec 29, 2022
Extending JAX with custom C++ and CUDA code

Extending JAX with custom C++ and CUDA code This repository is meant as a tutorial demonstrating the infrastructure required to provide custom ops in

Dan Foreman-Mackey 237 Dec 23, 2022
Offical code for the paper: "Growing 3D Artefacts and Functional Machines with Neural Cellular Automata" https://arxiv.org/abs/2103.08737

Growing 3D Artefacts and Functional Machines with Neural Cellular Automata Video of more results: https://www.youtube.com/watch?v=-EzztzKoPeo Requirem

Robotics Evolution and Art Lab 51 Jan 01, 2023
Implementation of TimeSformer, a pure attention-based solution for video classification

TimeSformer - Pytorch Implementation of TimeSformer, a pure and simple attention-based solution for reaching SOTA on video classification.

Phil Wang 602 Jan 03, 2023
Code repository for EMNLP 2021 paper 'Adversarial Attacks on Knowledge Graph Embeddings via Instance Attribution Methods'

Adversarial Attacks on Knowledge Graph Embeddings via Instance Attribution Methods This is the code repository to accompany the EMNLP 2021 paper on ad

Peru Bhardwaj 7 Sep 25, 2022