are there any available models to evaluate

[JH-Lam]

hi guy, thanks for your awesome effort. Now I'm try to predict some videos(Regression module) but found that some related models are not there. can you please
share some models to verify the performance of them( training data is optional).
thanks in advance and looking forward your reply!

[michael-mueller-git]

Hi JH-Lam. There is currently only a single checkpoint file available at the GitHub release tab. But this model is not really suitable for your intention. Since I don't have good hardware to train the models, I only did a short training period to see if the model converges during training. So if you want to evaluate the model you have to training your own model.
I think that the regression models are currently not the best approach to solve the problem (At least with my current implemented approach).

[JH-Lam]

Hi @michael-mueller-git , thanks and sorry for that I did not notice that link. I'll dive into it later. I thought some approaches before, eg. calculation based on landmarks, but I think regression should be the easiest one since there are many complex patterns on porn scenarios. any clues?

[michael-mueller-git]

I think regression should be the easiest one since there are many complex patterns on porn scenarios. any clues?

I agree with you on that. This was also my consideration when I developed the regression models. Additionally, you get the advantage that you don't need special labels for training, because existing funscripts + videos can be used as training material for this model types.
Please note that my code currently only handles VR videos. If you have powerful hardware for training it would be useful to increase the number of ConvLSTM layers in model1.

The main problem of the model 1 is that CNN does not encode the relative position of the different features and therefore large receptive fields are required to capture long-range dependencies within an image. Increasing the size of the convolution kernels or increasing the depth of the model can increase the representational capacity of the network, but this also loses the computational and statistical efficiency achieved by using a local convolutional structure. I think wee need some sort of self attention to solve this problem. See also: Transformers in Vision: A Survey and ConViT.

[JH-Lam]

Yep. It may be better to use RNN than CNN at sequence(temporal) data though I am not familiar with the RNN.
in addition, I want to add a binary classification to give a probability that indicates whether this regression is reasonable or not.
Thanks for detailed reply.

[JH-Lam]

Hi michael, pardon me if it's a bit late to raise a new thread here.

During these days ,I did a transfer learning on MobileNetV2 to train a custom regression for porn images(videos frames).
My scenarios:
-to detect how deep a toothbrush moved into the mouth
-only w/ range labels[0,100] but no other landmarks. 0 means fully moved into a mouth while 100 is out(or the tip of toothbrush is closed to lip)
-transform the predicted regression scalar to classification accuracy by : correct if and only if prediction belongs to [label-7,label+7]
-resize training samples to 224 to fit for mobileNetV2 input. so the object brushes are small enough to be called "small object detection"
-the training set is not even and some of them are mis-labeled.

(current distribution: 0-case is huge which is caused by data feature it-self while 100-case since I add many negative images to training set. it seems not reasonable if it labeled as the largest value of the range?

Results:
-low acc prediction on not-seem negative samples (ie. images which doesn't include brush actions), about 66%. and even worse in case of test set.
-train mae and valid mae are both closed to 18, while mse are closed 500. it seems very high.

What to do :
a. re-inspect all the training set to label them correctly
b. sample sparely the range into 10 times, eg. 0,10,..., 100 for gain more samples per label value. especially transform from regression to classification w/ these classes?
c. uses other models to handle the small object detection problem

Questions:

1. is it viable for my thinks described above
2. does the MobileNetV2 fit for small object detection in practice stated above?
3. what factors affect the model acc mainly?

thanks

[michael-mueller-git]

First of all, it's nice to see the progress you've made.

transform the predicted regression scalar to classification accuracy by : correct if and only if prediction belongs to [label-7,label+7]

if label is the output (predicted class) of your model, then this will not provide propper results. Because you have multiple correct classes assigned to an output. e.g. label 50 belongs to [43, 44, .. 50, .. 46, 57] and 51 belongs to [44, 44, .. 50, .. 46, 58], ... They overlap!

current distribution: 0-case is huge which is caused by data feature it-self while 100-case since I add many negative images to training set. it seems not reasonable if it labeled as the largest value of the range?

I think the many negative images do not influence the whole thing as much as the extremely unbalanced training material between 1..99

train mae und valid mae sind beide auf 18 geschlossen, während mse 500 geschlossen sind. das scheint sehr hoch.

The most important thing is that the loss curve of the validation data set decreases while training. (I recommend always use some sort of validation with Early Stopping)

uses other models to handle the small object detection problem

The main problem here is that you probably need very large models (which are then correspondingly slow) to get an acceptable accuracy. The main problem is: hat CNN does not encode the relative position of the different features and therefore large receptive fields are required to capture long-range dependencies within an image. Or directly translated: To determine the length of the toothbrush

is it viable for my thinks described above
does the MobileNetV2 fit for small object detection in practice stated above?

Not in my opinion. You probably won't get satisfactory results with such an approach.

what factors affect the model acc mainly?

From your description i assume you use somthing like the categorical_crossentropy for Loss. This does not make much sense in this case because otherwise the coherent distance of the regression values is discarded. You could try to add an Regesion Layer to the model and use MSE for Loss. If not already done.

In summary, the main problem with the approach is that it only makes a prediction locally per frame but does not take into account the previously predicted results. Since there are usually no jumps in a video (except video cuts). You could help the model by reintroducing the previously predicted regression values by using LSTMs after the feature extraction trough an CNN.

[JH-Lam]

Hi michael, thanks for your quick and detailed reply first.
I think there are some (may be wrong) statements described in my comment above:

1. the term 'toothbrush' is a avatar of 'blowjob' like terms which is not easy to describe in somewhere I asked before. sorry I forgot to restore it here :)
2. I dont want to use LSTMs related RNN techniques because it seemed slow when I tested your model before. in addition, I trained a simple MobileNet v2 (transfer learning in fact) long time before, it seemed good(only w/ one completely correct labeled funscript video) on valid set:
   
   ( it trends well though it's not absolutely correct. the training set has not negative images )
3. Yep I used regression on output instead of classification so I used MSE as loss while MAE is as metrics only in tensorflow. so statement 'transform the regression to accuracy' means I want to gain acc similar result but not train a classification model in fact. so I uses a workaround by doing : if prediction[i] in range [label[i]-k, label[i] + k] that means it's tolerant (correct)
4. No doubt , the valid loss curve is most important , but in my case it seems it tried hard to hard to go down further
   

Note the pos in funsripts are farely subjective (except 0 and 100) so I usually seem some mis-labeled ones in files.sometimes she maybe use hand to do that , or uses mouth sometimes, and some of the amplitude are explicitly really wrong. Maybe this is another cause to hard to train well?

5. "Since there are usually no jumps in a video (except video cuts)" . I think it's necessary to blend the pos which b/w the top and bottom pos even if I use skip-frames detection.
   eg. 0,10,...,100 , then when prediction come in '1,3,5,5, 10,11,20,5,0,...' , the resulted output(by round()) : 0,0,10,10,10,10, 20,10,0 => 0,20,0 . but I think this is tricky by program since there are more complex than this one. eg. big slope curve followed by small slope curve but you can't consider it as exactly one action .do you have any clues?
6. brainstorm: split sparely the regression range to 10x to do a classification instead is possible if I dont care the trivial changes b/w two ends ?
7. maybe I need to follow some age-face models to see if some inspirations are there . but I think it's harder to capture the porn action feature than face-texture feature ,right? DL too?

[michael-mueller-git]

2. This is understandable. LSTMs slow the whole thing down considerably.
3. As long as this statement is not included in the loss-function and thus the backpropagation algorithm, you can of course do this without any disadvantages.
4. OK the learning curve looks good. One reason why you can't get any further down is definitely the point you mentioned about subjective labels. Have you already tried to reduce the learning rate after x epochs, maybe the backpropagation algorithm will find a better minimum. But probably this will not achieve much more improvements. The impact of the subjective labels is probably too high.
5. For your model it is irrelevant because each frame is considered closed for itself. That was only a suggestion for the use of a recurrent network architecture which you do not want to use according to 2.
6. if you use a classification loss function that takes into account the distance of the classes from each other, this would be possible.
7. If you mean the face detection models. The predict the abs position in the frame which makes the training much better specified. Instead of labeling the frames you could label the top and bottom point of the good piece in the frames and use a face detection model architecture for prediction. This will probably result in much better than the current approach. That would be a slightly lighter approach compared to my Mask RCNN in this repository.

[JH-Lam]

4. yes I used :

lr_reduce = ReduceLROnPlateau(monitor='val_mae', factor=0.3, patience=5, verbose=1, cooldown=0, min_lr=0.5e-6)
but no luck. it seems no significant improvement

5. there is a fact that even if I predicted result well , but there are many actions generated via model to be saved in funscript files.
   
   
   so it's hard to know which actions should be ignored while others not. I think it's reasonable to do nothing in first iteration.
6. I saw a similar case which split age range from [0,100] to age groups ,says "0-2 to 0, 3-6 to 1, ...." to do a real classification model yesterday. https://github.com/CVxTz/face_age_gender
7. I mainly refer to 'small object detection' problem. the age model predicts age against face texture etc . so the texture is sure much small compared to a face. but I think it's easier than my job since face has explicit features to capture, eg. eyes, nose, mouth etc. in addition there are many complex scenarios w/o faces in fact, eg. sit down and up on a man's body but no faces at all. but I am not sure if DL agrees w/ me.

[JH-Lam]

Hi, after I dived into the code 'Regression/util/dataset.py' again, I noticed that :

1. therer are lables files which corresponds to each video , right?
2. and these files should be generated against uploaded funscripts by users respectively. so the actions in each label file is continuous b/w two frames labeled in funscript

if yes it's a good idea to generate even position values, no matter what positions the users made( but I used the uploaded funscripts' lables directly before)

[michael-mueller-git]

now that you mention the labels i noticed that i didn't put the script to convert the funscripts to the frame labels into git. I have now added these.

For each funscript (corresponding to an video) i creates labels for each frame through the python script. The labels are simply the regression values for the corresponding frame, where the funscript gets simply interpolated for each frame.

[JH-Lam]

got it , thanks.

some more questions:
a. why was it stuck in ffmpegstream.py at line 202 ( rojection = np.frombuffer(...) ) when I run label.py ?
b. why the behavior b/w training step and prediction step is not consitent? ie. the loss during training is calculated against each frame in a seq_len respectively, but in prediction the current action's pos is derived by the first value of current prediction. I think both should be consistent, eg. (pos[1]1,pos[2],...,pos[k]) ->(output) pos[k+1], so uses the last output is enough to compute the final loss.

train.py :
loss = criterion(out_pos, pos)
out_pos = model(frames);

test.py:
x = model(frames_tensor)
x = x.cpu().numpy()[0][0][0]
funscript.add_action(round(x * 100), round(frame_numer * (1000/fps)))

[michael-mueller-git]

a) there was a change in ffmpeg 5 which made the code not work anymore. Take a look at ffmpegstream.py in the Python-Funscript-Editor repository where I have fixed the issue. (I should probably use a git submodule in the future so that you don't have to copy it everywhere).
b) The current implementation only allow batch_size == 1. I think you mean the individual training sequence length. The problem is that i use the last seq_len frames for each frame. The idea was to use all values for the loss calculation to improve the convergence while training. But unfortunately has not helped much. The code is no longer completely consistent due to the experiments. Sorry for that. If you are lucky there is still something in the git history. But I think I published the code after my tests.

[JH-Lam]

Hi michael, I'm here again;)
I improved a bit performance these days. eg. give a clear start_frame to fast locate at the goal action since some funscripts uploaded are not started at first frame , but likely locate at any one.

The core thing I met currently is :
train_epoch_loss += loss.item()
It costs about 300ms in my gtx 1660 super with model1 ! I have no idea what happens in fact. but when I comment out 'loss.backward()' then it decrease at 100ms. so it seems likely this is BP related.

thanks

[michael-mueller-git]

Nice to see that you are still working on a funscript predictor model. I have currently stopped working on it due to other projects.

Theoretically you could remove train_epoch_loss += loss.item(). But then you would no longer have monitoring during the training and would not know if and how the model improves.

Yes loss.backward() is BP related and is required to train the model weights.