Skip to content

Commit 8af9f83

Browse files
updated
1 parent 7faf8af commit 8af9f83

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

Tutorials/custom_inference.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# Inference with a custom model
22

33
We have trained and evaluated the model, the next step is to see the performance of the model on unknown images. 
4-
We are going to test the model on the classes we have trained it on.
4+
We are going to test the model on the classes we have trained it on. If you are yet to download the nature's model trained on Nature dataset download it from [here](https://github.com/ayoolaolafenwa/PixelLib/releases/download/1.0.0/Nature_model_resnet101.h5)
55

66
*sample1.jpg*
77

Tutorials/custom_train.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ Implement custom training on your own dataset using PixelLib's Library. In just
77
Our goal is to create a model that can perform instance segmentation and object detection on butterflies and squirrels.
88
Collect images for the objects you want to detect and annotate your dataset for custom training. Labelme is the tool employed to perform polygon annotation of objects. Create a root directory or folder and within it create train and test folder. Separate the images required for training (a minimum of 300) and test.Put the images you want to use for training in the train folder and put the images you want to use for testing in the test folder. You will annotate both images in the train and test folder. Download [Nature's dataset](https://github.com/ayoolaolafenwa/PixelLib/releases/download/1.0.0/Nature.zip) used as a sample dataset in this article, unzip it to extract the images' folder. This dataset will serve as a guide for you to know how to organize your images.Ensure that the format of the directory of your own dataset directory is not different from it. Nature is a dataset with two categories butterfly and squirrel. There is 300 images for each class for training and 100 images for each class for testing i.e 600 images for training and 200 images for validation. Nature is a dataset with 800 images.
99

10-
## Labelme annotation tool is employed to perform polygon annotation of objects, read the tutorial here on how to annotate objects with Labelme.
10+
## Labelme annotation tool is employed to perform polygon annotation of objects, read this article on [medium](https://medium.com/@olafenwaayoola/image-annotation-with-labelme-81687ac2d077) on how to annotate objects with Labelme.
1111

1212
```
1313
Nature >>train>>>>>>>>>>>> image1.jpg
@@ -204,7 +204,7 @@ It shows that we are using *resnet50* for training.
204204

205205
# Model Evaluation
206206

207-
When we are done with training we should evaluate models with lowest validation losses. Model evaluation is used to access the performance of the trained model on the test dataset. Download the trained model from [here](https://github.com/ayoolaolafenwa/PixelLib/releases/download/1.0.0/Nature_model_resnet101.h5).
207+
When we are done with training we should evaluate models with lowest validation losses. Model evaluation is used to access the performance of the trained model on the test dataset.The model trained on Nature dataset and the dataset are available on the [release](https://github.com/ayoolaolafenwa/PixelLib/releases) of this github's repository. Download the trained model from [here](https://github.com/ayoolaolafenwa/PixelLib/releases/download/1.0.0/Nature_model_resnet101.h5).
208208

209209
```python
210210

0 commit comments

Comments
 (0)