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
A PyTorch Implementation of Gated Graph Sequence Neural Networks (GGNN)

A PyTorch Implementation of GGNN This is a PyTorch implementation of the Gated Graph Sequence Neural Networks (GGNN) as described in the paper Gated G

Ching-Yao Chuang 427 Dec 13, 2022
Fast, flexible and fun neural networks.

Brainstorm Discontinuation Notice Brainstorm is no longer being maintained, so we recommend using one of the many other,available frameworks, such as

IDSIA 1.3k Nov 21, 2022
Open source simulator for autonomous vehicles built on Unreal Engine / Unity, from Microsoft AI & Research

Welcome to AirSim AirSim is a simulator for drones, cars and more, built on Unreal Engine (we now also have an experimental Unity release). It is open

Microsoft 13.8k Jan 03, 2023
Invasive Plant Species Identification

Invasive_Plant_Species_Identification Used LiDAR Odometry and Mapping (LOAM) to create a 3D point cloud map which can be used to identify invasive pla

2 May 12, 2022
Pytorch implementation of Rosca, Mihaela, et al. "Variational Approaches for Auto-Encoding Generative Adversarial Networks."

alpha-GAN Unofficial pytorch implementation of Rosca, Mihaela, et al. "Variational Approaches for Auto-Encoding Generative Adversarial Networks." arXi

Victor Shepardson 78 Dec 08, 2022
MediaPipe is a an open-source framework from Google for building multimodal

MediaPipe is a an open-source framework from Google for building multimodal (eg. video, audio, any time series data), cross platform (i.e Android, iOS, web, edge devices) applied ML pipelines. It is

Bhavishya Pandit 3 Sep 30, 2022
TinyML Cookbook, published by Packt

TinyML Cookbook This is the code repository for TinyML Cookbook, published by Packt. Author: Gian Marco Iodice Publisher: Packt About the book This bo

Packt 93 Dec 29, 2022
Code for the paper "Controllable Video Captioning with an Exemplar Sentence"

SMCG Code for the paper "Controllable Video Captioning with an Exemplar Sentence" Introduction We investigate a novel and challenging task, namely con

10 Dec 04, 2022
Implementation of "Debiasing Item-to-Item Recommendations With Small Annotated Datasets" (RecSys '20)

Debiasing Item-to-Item Recommendations With Small Annotated Datasets This is the code for our RecSys '20 paper. Other materials can be found here: Ful

Microsoft 34 Aug 10, 2022
A generalist algorithm for cell and nucleus segmentation.

Cellpose | A generalist algorithm for cell and nucleus segmentation. Cellpose was written by Carsen Stringer and Marius Pachitariu. To learn about Cel

MouseLand 733 Dec 29, 2022
TaCL: Improving BERT Pre-training with Token-aware Contrastive Learning

TaCL: Improving BERT Pre-training with Token-aware Contrastive Learning Authors: Yixuan Su, Fangyu Liu, Zaiqiao Meng, Lei Shu, Ehsan Shareghi, and Nig

Yixuan Su 79 Nov 04, 2022
It is a system used to detect bone fractures. using techniques deep learning and image processing

MohammedHussiengadalla-Intelligent-Classification-System-for-Bone-Fractures It is a system used to detect bone fractures. using techniques deep learni

Mohammed Hussien 7 Nov 11, 2022
Code for database and frontend of webpage for Neural Fields in Visual Computing and Beyond.

Neural Fields in Visual Computing—Complementary Webpage This is based on the amazing MiniConf project from Hendrik Strobelt and Sasha Rush—thank you!

Brown University Visual Computing Group 29 Nov 30, 2022
Analysis of rationale selection in neural rationale models

Neural Rationale Interpretability Analysis We analyze the neural rationale models proposed by Lei et al. (2016) and Bastings et al. (2019), as impleme

Yiming Zheng 3 Aug 31, 2022
4th place solution to datafactory challenge by Intermarché.

Solution to Datafactory challenge by Intermarché. 4th place solution to datafactory challenge by Intermarché. The objective of the challenge is to pre

Raphael Sourty 11 Mar 19, 2022
The Pytorch implementation for "Video-Text Pre-training with Learned Regions"

Region_Learner The Pytorch implementation for "Video-Text Pre-training with Learned Regions" (arxiv) We are still cleaning up the code further and pre

Rui Yan 0 Mar 20, 2022
Deep Learning Theory

Deep Learning Theory 整理了一些深度学习的理论相关内容,持续更新。 Overview Recent advances in deep learning theory 总结了目前深度学习理论研究的六个方向的一些结果,概述型,没做深入探讨(2021)。 1.1 complexity

fq 103 Jan 04, 2023
[ICCV 2021] Code release for "Sub-bit Neural Networks: Learning to Compress and Accelerate Binary Neural Networks"

Sub-bit Neural Networks: Learning to Compress and Accelerate Binary Neural Networks By Yikai Wang, Yi Yang, Fuchun Sun, Anbang Yao. This is the pytorc

Yikai Wang 26 Nov 20, 2022
Code for Graph-to-Tree Learning for Solving Math Word Problems (ACL 2020)

Graph-to-Tree Learning for Solving Math Word Problems PyTorch implementation of Graph based Math Word Problem solver described in our ACL 2020 paper G

Jipeng Zhang 66 Nov 23, 2022
Dilated Convolution for Semantic Image Segmentation

Multi-Scale Context Aggregation by Dilated Convolutions Introduction Properties of dilated convolution are discussed in our ICLR 2016 conference paper

Fisher Yu 764 Dec 26, 2022