Skip to content

Commit

Permalink
Update Readme
Browse files Browse the repository at this point in the history
  • Loading branch information
ajaichemmanam committed Sep 1, 2020
1 parent 6ff487d commit af32273
Showing 1 changed file with 14 additions and 19 deletions.
33 changes: 14 additions & 19 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,27 +7,8 @@ This repo uses Centernet and Conditional Convolutions for Instance Segmentation
## Installation

Please refer to [INSTALL.md](readme/INSTALL.md) for installation instructions.

This repo supports both CPU and GPU Training and Inference.

## Pre-Trained Models

Pre-Release : [Google Drive](https://drive.google.com/drive/folders/1Uw0ucRLpyyHT0pGW2N0o5BcYdSfdNYyC?usp=sharing)

Download the recent model, copy to exp/ctseg/coco_dla_1x/

Rename as model_last.pth

```
python3 demo.py ctseg --exp_id coco_dla_1x --keep_res --resume --demo ../data/coco/val2017
```

Note: Model is still training. The initial model may give poor results.

Will update in the drive once the training finishes.

## Setup
```
git clone --recurse-submodules https://github.com/ajaichemmanam/CenterSeg.git
Expand All @@ -49,6 +30,20 @@ cd src/lib/external
python3 setup.py build_ext --inplace
```

## Pre-Trained Models

Pre-Release : [Google Drive](https://drive.google.com/drive/folders/1Uw0ucRLpyyHT0pGW2N0o5BcYdSfdNYyC?usp=sharing)

Download the most recent model (model_last_e101.pth), copy to exp/ctseg/coco_dla_1x/

Rename as model_last.pth

```
python3 demo.py ctseg --exp_id coco_dla_1x --keep_res --resume --demo ../data/coco/val2017
```

Note: Model is not completely trained (101 Epochs only). Will update later.

#### Training

###### For GPU
Expand Down

0 comments on commit af32273

Please sign in to comment.