1
$\begingroup$

I want to create a ship detection classifier from a dataset that is formed by 4000 photos(3072*2048).

But the dataset that i currently have is not labeled so i can feed it to a cnn.So i want to cluster this dataset to 2 labels(or 2 directories) ship and no_ship.I tried running k-means but the results were dissapointing.Is some other more functional way to do this?

$\endgroup$

3 Answers 3

1
$\begingroup$

In addition to the answers provided, you can:

1.) Train jointly a CNN (or Autoencoder) with clustering on your data. (DCN, kmeansNet,..)

2.) Pretrain a CNN using self-supervision on your data. (Have a look into the vast self-supervision literature, e.g. this work).

3.) Use an alternating scheme to train a CNN classifier on soft-labels provided by a clustering algorithm, e.g k-means (e.g. this work).

$\endgroup$
1
  • $\begingroup$ May I kindly draw your attention to my question in this regard? It will be highly appreciated. $\endgroup$
    – Mario
    Dec 3, 2020 at 15:43
0
$\begingroup$

Here's what you could try.

  1. Find a pre-trained network which is capable of detecting ships (An example could be a network trained on ImageNet). You will only need the layers before the Softmax() layer or after the Flatten() layer
  2. If there are multiple types of ships you want to detect, I would pass multiple images of ships and non-ships into the network. For each ship/non-ship image, you will obtain a 1-D feature embedding. You could then average out the embeddings of all the ship & non-ship images you choose. What this tells you is that pictures with/without ships should have an embedding that looks like this.
  3. Lastly, pass each image in your unlabelled dataset through the network and use a distance metric to see whether it is closer to the embedding that represents ships or the one that represents non-ships. You could use different metrics as shown here: https://dataaspirant.com/2015/04/11/five-most-popular-similarity-measures-implementation-in-python/
$\endgroup$
2
  • $\begingroup$ thank you for your answer!! Can you provide me with some code or a good paper to start with? Furthermore,where can i find a pre-trained network for ships? $\endgroup$
    – scoute21
    Apr 9, 2020 at 12:56
  • $\begingroup$ Refer to this link for some pretrained models you can use: keras.io/applications. There is a header: "Fine-tune InceptionV3 on a new set of classes" which shows you how you can remove the last few layers as you only need the feature maps and not the final prediction on the classes. From there, you can then use the model to predict on images with ships to get an average embedding for the 2 scnarios, and then match your unlabelled images based on the distance metric you use. $\endgroup$ Apr 9, 2020 at 17:14
0
$\begingroup$

First of all, keep this in mind:

After all, if it was easy to do this without any labels, then, what would be the point of needing the labels in the first place?

I can see two options:

  1. Use a pre-trained image classifier to represent your images

As Vincent Young suggests, you can find pre-trained networks which have been trained on similar detection tasks. ModelZoo is a good place to find pre-trained networks for the framework you are using.

  1. Try mean-shift instead of K-Means

K-Means is straight forward but has some flow. It doesn't deal well with clusters of uneven size and will learn towards creating circular clusters due to Euclidean distance.

Mean-shift can deal with arbitrary feature spaces and can use arbitrary kernel functions. You may not end up with 2 clusters, but you may be able to find useful clusters regardless. On this note, if you try using more than 2 clusters with K-Means, you may find some clusters being "pure" (containing a single class) while some may be mixed. These pure clusters can be a good start.

I wrote a chapter on Mean Shift on my website, including other resources, if you want to read it.

enter image description here

$\endgroup$
1
  • $\begingroup$ May I kindly draw your attention to my question in this regard? It will be highly appreciated. $\endgroup$
    – Mario
    Dec 3, 2020 at 15:41

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.

Not the answer you're looking for? Browse other questions tagged or ask your own question.